- Python 3.8.0
- PyTorch 1.10.1+cu12.2
- Transformers 4.27.4
The code has been tested on Ubuntu 20.04 using a single RTX 3090 GPU.
- You can download the IEMOCAP and EmoryNLP dataset, extracted utterances we used from: here
- Run:
python EmoTransNet/EmoTransKG_Construction.py
- Please download the IEMOCAP and EmoryNLP datasets and put them in the data folder. And we utilize the codes from here to extract utterance or event features.
- Run our EmoTransNet model:
python EmoTransNet/EmoTransNet.py --lr 2e-5 --l2 3e-4 --dropout 0.5 --batch_size 64 --transformer_layers 6 --epoches 100