Code for "Like hiking? You probably enjoy nature: Persona-grounded Dialog with Commonsense Expansions"

Like hiking? You probably enjoy nature: Persona-grounded Dialog with Commonsense Expansions

Bodhisattwa Prasad Majumder, Harsh Jhamtani, Taylor Berg-Kirkpatrick, Julian McAuley

Published at EMNLP 2020.

Data

Huggingface cleaned dataset

wget https://s3.amazonaws.com/datasets.huggingface.co/personachat/personachat_self_original.json

(Optional) Personachat Download: http://parl.ai/downloads/personachat/personachat.tgz

COMET-expanded PersonaChat

You can download PersonaChat dataset with added COMET expansions from here.

Training

The training script is here. Please download the data (mainly COMET expansions) before training. The training may require 4 2080Tis or higher.

Citation

If you find compac useful for your research, please cite our paper:

@inproceedings{MajumderJBM20,
  author    = {Bodhisattwa Prasad Majumder and
               Harsh Jhamtani and
               Taylor Berg{-}Kirkpatrick and
               Julian J. McAuley},
  title     = {Like hiking? You probably enjoy nature: Persona-grounded Dialog with Commonsense Expansions},
  booktitle = {EMNLP},
  year      = {2020},
  url       = {https://doi.org/10.18653/v1/2020.emnlp-main.739},
  doi       = {10.18653/v1/2020.emnlp-main.739},
}