Home

Awesome

MedDG

This is the code for the following paper:

MedDG: an entity-centric medical consultation dataset for entity-aware medical dialogue generation

Wenge Liu, Jianheng Tang, Yi Cheng, Wenjie Li, Yefeng Zheng, Xiaodan Liang;

Requirement

pip install allenlp==0.9

Usage

For the task of topic prediction (e.g. run the LSTM baseline):

cd topic_predict
allennlp train ../config/topic_predict/lstm_baseline.json --include-package basline -s tmp/lstm1

For the task of response generation (e.g. run the seq2seq baseline):

cd generation
allennlp train ../config/generation/seq2seqattn.json --include-package seq2seq -s tmp/seq2seq1

Dataset

https://drive.google.com/drive/folders/109WnXlNhmqttxYwb4EEscYGcA_-eq8Eg?usp=sharing