NUSTM / ECPE-2D

ECPE-2D: Emotion-Cause Pair Extraction based on Joint Two-Dimensional Representation, Interaction and Prediction

This repository contains the code for our ACL 2020 paper:

Zixiang Ding, Rui Xia, Jianfei Yu. ECPE-2D: Emotion-Cause Pair Extraction based on Joint Two-Dimensional Representation, Interaction and Prediction. ACL 2020. [pdf]

Please cite our paper if you use this code.

Dependencies

• Python 2 (tested on python 2.7.15)
• Tensorflow 1.13.1
• Bert (The pretrained bert model "BERT-Base, Chinese" is required. Our model is built based on this implementation: https://github.com/google-research/bert)

Usage

• For ECPE-2D(Indep, Inter-CE, Inter-EC) models, run:

  • python main.py

• For ECPE-2D(Inter-EC(BERT)) models, run:

  • python Bert_main.py