MedDG

This is the code for the following paper:

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

pip install allenlp==0.9

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

Provide feedback