Skip to content

A theano implementation of the neural conversational model

License

Notifications You must be signed in to change notification settings

QingGeGe/Seq2Seq-Chatbot

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

40 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

A theano implementation of the neural conversational model

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%