这是一个用Python+Tensorflow实现的聊天机器人程序,使用seq2seq模型训练。示例所用训练数据集是IMDB600多部电影中的英文台词对话部分,训练时间为3天左右(2012款MacBook Pro i7),目前仅支持英文。另外程序包含一个简单的Python+Flask WebUI,并实现了微信公众号对接功能。
操作说明
execute.py为Python主程序,程序有三种模式:训练、测试和服务,可通过修改配置文件seq2seq.ini
来改变模式:
- 选择运行模式:
- 训练模式:
mode = train
- 测试模式:
mode = test
- 训练模式:
- 前台进程方式启动程序:
python execute.py
- WEB服务方式启动程序:
python webui/app.py
- 需预先安装Flask环境:
setup.sh/requirements.txt
- 通过脚本启动后台服务:
sh webui/startup.sh
- 需预先安装Flask环境: