Empathetic Dialogue Generation via Sensitive Emotion Recognition and Sensible Knowledge Selection (Findings of EMNLP 2022)

This repository is the implementation of our EMNLP 2022 Paper: Empathetic Dialogue Generation via Sensitive Emotion Recognition and Sensible Knowledge Selection

The paper is available at https://arxiv.org/abs/2210.11715.

Usage

Dependencies

Install the required libraries

pip install -r requirements.txt 

• Download Pretrained GloVe Embeddings and save it in /vectors.

• The preprocessed dataset is saved as /data/ED/dataset_preproc.p. If you want to create the dataset yourself or change the knowledge types generated by COMET, delete this file, download the COMET checkpoint and place it in /data/ED/Comet. The preprocessed dataset would be generated after the training script.

Training

python main.py --cuda --save_path save/your_dir

Testing

python main.py --cuda --test --save_path save/your_dir --model_path save/dir_save/SEEK_XXXX.XXX

---

Please be free to contact with us via wanglanrui@iie.ac.cn.