This repository is the official implementation of [Conversational Emotion-Cause Pair Extraction with Guided Mixture of Experts].
- Python 3.9.7
- PyTorch 1.11.0
- NumPy 1.20.3
- sickit-learn 0.24.2
- transformers 4.11.3
- CUDA 11.4
- tqdm 4.62.3
To train (or test) the model in the paper, run this command:
python main.py
We describes the meaning of arguments in main.py.
The dataset used in this paper is RECCON dataset. See this link for more information.
TBD