miguelcl/seq2seq-chatbot

Chatbot in 200 lines of code using TensorLayer

Seq2Seq Chatbot

This is a 200 lines implementation of Twitter/Cornell-Movie Chatbot, please read the following references before you read the code:

• Practical-Seq2Seq
• The Unreasonable Effectiveness of Recurrent Neural Networks
• Understanding LSTM Networks (optional)

Prerequisites

• Python
• TensorFlow >= 1.2
• TensorLayer >= 1.6.3

Model

Training

python main.py --batch-size 32 --num-epochs 50 -lr 0.001

Inference

python main.py --inference

Results

Query > happy birthday have a nice day
 > thank you so much
 > thank babe
 > thank bro
 > thanks so much
 > thank babe i appreciate it
Query > donald trump won last nights presidential debate according to snap online polls
 > i dont know what the fuck is that
 > i think he was a racist
 > he is not a racist
 > he is a liar
 > trump needs to be president