Chinese-Chatbot-base-on-Transformer

This is a chiese chatbot base on the Transformer mechanism(attention is all you need), this project is base on changwookjun/Transformer. You can change the hyperparameters in config.py.

prerequestion

Python 3.6.6
tensorflow 1.11
jieba
pandas
sklearn

How to training

python main_ch.py

How to predict

python predict.py