Skip to content
In this demo code, we implement Tensorflows Sequence to Sequence model to train a chatbot on the Cornell Movie Dialogue dataset.The dataset has been in the project. After training for a few hours, the bot is able to hold a fun conversation.
Python Shell
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.idea tensorflow-seq2seq-chatbot Jan 9, 2017
__pycache__ tensorflow-seq2seq-chatbot Jan 9, 2017
data tensorflow-seq2seq-chatbot Jan 9, 2017
.gitignore
README.md
__init__.py tensorflow-seq2seq-chatbot Jan 9, 2017
data_utils.py tensorflow-seq2seq-chatbot Jan 9, 2017
env_setup.sh
execute.py tensorflow-seq2seq-chatbot Jan 9, 2017
neuralconvo.ini tensorflow-seq2seq-chatbot Jan 9, 2017
seq2seq.ini tensorflow-seq2seq-chatbot Jan 9, 2017
seq2seq_model.py
seq2seq_serve.ini tensorflow-seq2seq-chatbot Jan 9, 2017

README.md

#Tensorflow Seq2Seq Chatbot Overview

In this demo code, we implement Tensorflows Sequence to Sequence model to train a chatbot on the Cornell Movie Dialogue dataset.The dataset has been in the project. After training for a few hours, the bot is able to hold a fun conversation.

Example QA

where are you? I ' m here .

how is going? Fine .

Can you think of it? I don ' t think so .

who is your father I ' m not sure .

what is time now? I don ' t know .

what the hell What ?

What are you doing now? I ' m going to get out .

Dependencies

Use pip to install any missing dependencies

Usage

Firstly, create an empty directory named woking_dir in the project. To train the bot, edit the seq2seq.ini file so that mode is set to train like so

mode = train

then run the code like so

python execute.py

To test the bot during or after training, edit the seq2seq.ini file so that mode is set to test like so

mode = test

then run the code like so

python execute.py

###Due date: December 8th

Also see this issue, some people have found this discussion helpful https://github.com/llSourcell/tensorflow_chatbot/issues/3

Credits

Credit for the vast majority of code here goes to suriyadeepan. I've merely created a wrapper to get people started.

You can’t perform that action at this time.