Realization for paper: "Learning to Select Knowledge for Response Generation in Dialog Systems", Rongzhong Lian, Min Xie, Fan Wang, Jinhua Peng, Hua Wu, IJCAI 2019 (PostKS)
- PyTorch 0.4.1+
- Python 3.5+
Put train/valid/test data files in data/ folder with the same prefix:
$prefix.train$prefix.valid$prefix.test
Adding embeddings if you need a warm start
We provide only a small size dataset for demonstration
python run_seq2seq.py : Seq2Seq
python run_knowledge.py : PostKS with HGFU
python run_seq2seq.py --test --ckpt $model_state_file