Branch: master
Find file History

README.md

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.

Examples

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':

python projects/personachat/scripts/kvmemnn_interactive.py

Evaluate pre-trained profile memory model trained to optimize f1 metric on Persona-Chat using persona 'self':

python projects/personachat/scripts/profilememory_eval_f1.py

Train a profile memory model on Persona-Chat using persona 'self':

bash projects/personachat/scripts/profilememory_train.sh