/Inspired

Primary LanguagePython

Inspired

Dataset and code for the EMNLP 2020 paper "INSPIRED: Toward Sociable Recommendation Dialogue Systems".

Dataset

Annotated dialogs (train.tsv, dev.tsv, test.tsv) is available under data/dialog_data.

Survey result is under data/survey_data.

Movie database is in data/movie_database.tsv.

Code

We provide the code for our dialog model.

  • data_preprocessing.py
  • training.py
  • interactive.py

Model Weights

Model weights can be downloaded here

Citation

@inproceedings{hayati-etal-2020-inspired,
    title = "INSPIRED: Toward Sociable Recommendation Dialog Systems",
    author = "Hayati, Shirley Anugrah  and Kang, Dongyeop  and Zhu, Qingxiaoyang  and Shi, Weiyan  and Yu, Zhou",
    booktitle = "Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP)",
    month = nov,
    year = "2020",
    address = "Online",
    publisher = "Association for Computational Linguistics",
    url = "https://www.aclweb.org/anthology/2020.emnlp-main.654",
    pages = "8142--8152",
}

Contact

Shirley Anugrah Hayati - sweetpineappleforest@gmail.com