Source code for the paper "Elementary Discourse Units with Sparse Attention for Multi-label Emotion Classification"
python 3.7
torch 1.9.0
Other packages can be installed via:
pip install -r requirements.txt
- SemEval18: Download EDU segmentation toolkits SegBot and run data/preprocess.py.
- Ren_CECps: The use of this data set requires permission from the author. Download EDU segmentation toolkits (here) and run parser.py.
python scripts/train.py
python scripts/test.py