This repository has stopped maintenance, new project links: pytorch-chatbot

• Python 2.7
• Pytorch 0.12

• Cornell Movie Dialogue

python preprocess.py

This script will create file dialogue_corpus.txt in ./data directory.

python train.py

The hyperparameters of model define in configuration file config.json.
In my local environment(GTX1060), training model need about four hours.

python chatbot.py

> hi .
bot: hi .
> what's your name ?
bot: jacob .
> how are you ?
bot: fine .
> where are you from ?
bot: up north .
> are you happy today ?
bot: yes .

The chatbot can answer these simple questions, but in most cases it is a silly bot.

• PyTorch documentation
• seq2seq-translation
• tensorflow_chatbot
• Cornell Movie Dialogs Corpus