PersonaChat

This project was archived on 2020-08-01. The models and scripts have not been tested in a long time. You may continue to use the data without issue (--task personachat). If you need to use the model or scripts, please rewind to the personachat tag.

Old README

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. Also note that the Profile Memory model has been deprecated and removed from this codebase; for generative models trained on the ConvAI2 dataset you can see that project folder.

Examples

Look at the PersonaChat data:

parlai display_data --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