Chatbot in 200 lines of code using TensorLayer
Switch branches/tags
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data check 1.8.1 works Mar 11, 2018
.gitignore [RE] Inference mode and TF compatibility (#16) (#20) Sep 28, 2018
README.md Update README.md (#33) Dec 14, 2018
main.py remove exception handler Oct 6, 2018
requirements.txt [RE] Inference mode and TF compatibility (#16) (#20) Sep 28, 2018

README.md

Seq2Seq Chatbot

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

Prerequisites

Model


Training

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

Inference

python main.py --inference-mode

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