Skip to content

This repository provides the code and dataset for the work published in the paper - Modeling Label Semantics for Predicting Emotional Reactions

Notifications You must be signed in to change notification settings

StonyBrookNLP/emotion-label-semantics

Repository files navigation

(Work in progress) Modeling-Label-Semantics-for-Predicting-Emotional-Reactions

Code for the paper - Modeling Label Semantics for Predicting Emotional Reactions - https://arxiv.org/pdf/2006.05489.pdf. This work explores the use different ways of utilizing label semantics techniques for improving multi-label emotion prediction in short commonsense stories:

GitHub Logo Multi-label emotion reactions in ROC Stories (Rashkin et al.)

Running experiments

Training a model

This section covers training a classification model for the Semi-supervision model and the baseline models (refer to Table 1 in the paper). The model names and corresponding main file names are mentioned below:

  1. BERT - main_bert_base.py
  2. LEAM w/ Glove - main_leam_glove_label.py
  3. LEAM w/ BERT Features - main_leam_bert_label.py
  4. BERT + Labels as Input - main_bert_label.py
  5. Learned Correlations - main_bert_learned_correlation.py
  6. Semi-supervision - main_bert_soft_semi_supervision.py

Run the following command for training each of above listed models with the correct main file:

python main_bert_soft_semi_supervision.py --do_train --save_best --do_lower_case --monitor valid_loss --mode min --epochs 20 --train_batch_size 8

Code citations

This work wouldn't be possible without the code shared in the following repositories, which were used heavily in this work

About

This repository provides the code and dataset for the work published in the paper - Modeling Label Semantics for Predicting Emotional Reactions

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages