This code is a python re-implementation of the persona-based neural conversation model proposed by Jiwei Li, Michel Galley, Chris Brockett, Georgios P. Spithourakis, Jianfeng Gao, Bill Dolan:

J.Li, M.Galley, C.Brockett, J.Gao and B.Dolan. "A persona-based neural conversation model". ACL2016.

Thank them for their great contributions :)

The code was used in our project of "Automatic Evaluation of Neural Personality-based Chatbot":

Y.Xing, R.Fernández. "Automatic Evaluation of Neural Personality-based Chatbot". INLG 2018.

Please go to "https://github.com/jiweil/Neural-Dialogue-Generation" for the processed data provided by Jiwei Li.

Similar to the original Lua codes, run

python train.py

The parameters could be changed in params.py.

After training, trained models are stored in /save.

Similar to the original Lua codes, run

python decode.py

The parameters could be changed in decode_params.py.

After decoding, the outputs are stored in /outputs.

Please refer to https://github.com/jiweil/Neural-Dialogue-Generation for more details.