KERS: A Knowledge-Enhanced Framework for Recommendation Dialog Systems with Multiple Subgoals

We provide the code for our dialog model.

• train_subgoal_generation.py: to train subgoal generation model
• run_predict_subgoal.py: to predict subgoal
• train_knowledge_generation.py: to train knowledge generation model
• run_predicate_knowledge.py: to predict knowledge
• run_KERS.py: to train KERS model
• run_generation_KERS.py: to generate response

Citation

@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",
}