0%

seq2seq 模型聊天机器人

这是一个用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

示例效果

easybot2
easybot3

参考资料

Jack wechat
微信公众号:EZApps
喜欢就好 赞赏其次