Skip to content

A theano implementation of the neural conversational model

License

Notifications You must be signed in to change notification settings

damientseng/Seq2Seq-Chatbot

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

A seq2seq Chatbot

Code Grade

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.

About

A theano implementation of the neural conversational model

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages