Skip to content

Code for Towards Knowledge-Enriched Conversational Recommendation System

License

Notifications You must be signed in to change notification settings

stephenliu0423/KECRS

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

KECRS

Towards Knowledge-Enriched Conversational Recommendation System.

Prerequisites

  • Python 3.6
  • PyTorch 1.4.0
  • Torch-Geometric 1.4.2

Getting Started

Installation

Clone this repo.

git clone https://github.com/xiaolan98/KECRS.git
cd KECRS/parlai/task/crs/

Dataset

All the data are in * ./KECRS/data/crs/ *folder

  • ReDial dataset
  • The Movie Domain Knowledge Graph, TMDKG

Training

To train the recommender part, run:

python train_kecrs.py

To train the dialog part, run:

python train_transformer_rec.py

Logging

TensorBoard logs and models will be saved in saved/ folder.

Evaluation

All results on testing set will be shown after training.

Discussion

If you have difficulties to get things working in the above steps, please let us know.

Cite

Please cite our paper if you use this code in your own work:

@inproceedings{zhang-etal-2022-toward,
    title = "Toward Knowledge-Enriched Conversational Recommendation Systems",
    author = "Zhang, Tong  and
      Liu, Yong  and
      Li, Boyang  and
      Zhong, Peixiang  and
      Zhang, Chen  and
      Wang, Hao  and
      Miao, Chunyan",
    booktitle = "Proceedings of the 4th Workshop on NLP for Conversational AI",
    month = may,
    year = "2022",
    address = "Dublin, Ireland",
    publisher = "Association for Computational Linguistics",
    url = "https://aclanthology.org/2022.nlp4convai-1.17",
    doi = "10.18653/v1/2022.nlp4convai-1.17",
    pages = "212--217",
    }

About

Code for Towards Knowledge-Enriched Conversational Recommendation System

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%