This directory contains example training and interactive scripts for some of the models used in the paper Personalizing Dialogue Agents: I have a dog, do you have pets too?. If you are interested in the Mechanical Turk tasks used to collect the Persona-Chat dataset used in this paper, you can view them here.
Note that the dataset for the ConvAI2 competition is larger than the Persona-Chat dataset, so we are computing new baselines for that competition. See the baselines for that competition here.
Look at the PersonaChat data:
python examples/display_data.py --task personachat --datatype train
Interact with a pre-trained Key-Value Memory Net model trained on Persona-Chat using persona 'self':
Evaluate pre-trained profile memory model trained to optimize f1 metric on Persona-Chat using persona 'self':
Train a profile memory model on Persona-Chat using persona 'self':