mjc14 / seq2seq-chatbot

A theano implementation of the neural conversational model

A seq2seq Chatbot

A chatbot based on seq2seq model, implemented by Theano, based on the Google paper A Neural Conversational Model.

The training dataset can be found here: Cornell Movie--Dialogs Corpus

======== ##How to Use First, have theano installed.

Download the dataset above, make a new folder called data for it.

To prepare the data for training, run make_convs.py.

Next run build_model.py to train:

$THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python build_model.py

Note that the last few lines are commented out, which illustrate how to make a Chatbot instance with a model instance, and to talk to the chatbot instance by simply passing it a string.