/Emotion-Knowledge-Graph-EmoTransKG

Primary LanguagePython

Requirements

  • 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.

Constructing EmoTransKG

  1. You can download the IEMOCAP and EmoryNLP dataset, extracted utterances we used from: here
  2. Run:
python EmoTransNet/EmoTransKG_Construction.py

EmoTransNet: Training EmoTransKG

  1. 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.
  2. 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

Conversational Emotion Recognition with EmoTransNet

Controlling variables in conversation