ChatGirl 一个基于 TensorFlow Seq2Seq 模型的聊天机器人[中文文档]
简介
简单地说就是该有的都有了,但是总体跑起来效果还不好。
还在开发中,它工作的效果还不好。但是你可以直接训练,并且运行。
包含预处理过的 twitter 英文数据集,训练,运行,工具代码,可以运行但是效果有待提高。
数据集
Twitter 数据集:
训练
你需要新建一个 model 文件夹来保存训练完的模型
运行这个文件来训练模型
Train_Model.py
运行
运行这个文件来运行模型
Run_model.py
工具集
w2idx:单词到 id ,用于下一步将单词转化为向量。
idx2w:id 到 单词,用于将 id 转换为单词。
hello.py 中有两个函数的 demo。
结果
Result.md(部分输出)
博客
http://www.tensorflownews.com/
开发计划
- 优化数据集
- 优化