We provide the code for our dialog model.
train_subgoal_generation.py
: to train subgoal generation modelrun_predict_subgoal.py
: to predict subgoaltrain_knowledge_generation.py
: to train knowledge generation modelrun_predicate_knowledge.py
: to predict knowledgerun_KERS.py
: to train KERS modelrun_generation_KERS
.py: to generate response
@inproceedings{zhang-etal-2021-kers-knowledge,
title = "{KERS}: A Knowledge-Enhanced Framework for Recommendation Dialog Systems with Multiple Subgoals",
author = "Zhang, Jun and
Yang, Yan and
Chen, Chencai and
He, Liang and
Yu, Zhou",
booktitle = "Findings of the Association for Computational Linguistics: EMNLP 2021",
month = nov,
year = "2021",
address = "Punta Cana, Dominican Republic",
publisher = "Association for Computational Linguistics",
url = "https://aclanthology.org/2021.findings-emnlp.94",
doi = "10.18653/v1/2021.findings-emnlp.94",
pages = "1092--1101",
}