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
The following command automatically performs training and evaluation for all data.
python train.py
The dataset used in this paper is RECCON dataset. See this link for more information.
DongJin Jeong and JinYeong Bak. 2023. Conversational Emotion-Cause Pair Extraction with Guided Mixture of Experts. In Proceedings of the 17th Conference of the European Chapter of the Association for Computational Linguistics, pages 3288–3298, Dubrovnik, Croatia. Association for Computational Linguistics.