ChatGirl 一個基於 TensorFlow Seq2Seq 模型的聊天機器人
簡介
還在開發中,它工作的效果還不好。但是你可以直接訓練,並且運行。
包含預處理過的 twitter 英文數據集,訓練,運行,工具代碼,可以運行但是效果有待提高。
數據集
Twitter 數據集:
https://github.com/suriyadeepan/datasets
訓練
你需要新建一個 model 文件夾來保存訓練完的模型
運行這個文件來訓練模型
Train_Model.py
運行
運行這個文件來運行模型
Run_model.py
工具集
w2idx:單詞到 id ,用於下一步將單詞轉化為向量。
idx2w:id 到 單詞,用於將 id 轉換為單詞。
hello.py 中有兩個函數的 demo。
結果
Result.md(部分輸出)
博客
http://www.tensorflownews.com/
開發計劃
- 優化數據集
- 優化模型
[under developing]
TensorFlowNews 項目推薦:
基於 TensorFlow 的神經網路三大主流模型卷積神經網路,循環神經網路,序列到序列模型 TensorFlowNews 都原創了實戰項目,包含模型,代碼,數據集,工具集,歡迎 star :
- TensorFlow CNN Model Project:
FaceRank - Rank Face by CNN Model based on TensorFlow (add keras version). FaceRank-人臉打分基於 TensorFlow (新增 Keras 版本) 的 CNN 模型(可能是最有趣的 TensorFlow 中文入門實戰項目)
https://github.com/fendouai/FaceRank
- TensorFlow LSTM Model Project:
一個比特幣交易機器人基於 Tensorflow LSTM 模型,僅供娛樂。 A Bitcoin trade robot based on Tensorflow LSTM model.Just for fun.
https://github.com/TensorFlowNews/TensorFlow-Bitcoin-Robot
- TensorFlow Seq2Seq Model Project:
ChatGirl is an AI ChatBot based on TensorFlow Seq2Seq Model.ChatGirl 一個基於 TensorFlow Seq2Seq 模型的聊天機器人。(包含預處理過的 twitter 英文數據集,訓練,運行,工具代碼,可以運行但是效果有待提高。)
https://github.com/fendouai/ChatGirl
學習流程
如果看到這裡有很多不懂的話,建議:
* Hello World
https://zhuanlan.zhihu.com/p/27963600
* 基本概念
https://zhuanlan.zhihu.com/p/27986689
* 卷積神經網路
https://zhuanlan.zhihu.com/p/28161292
* 訓練好模型參數的保存和恢復代碼
https://zhuanlan.zhihu.com/p/27912379
* TensorFlowNews 專欄
https://zhuanlan.zhihu.com/TensorFlownews
* TensorFlowNews 博客
http://www.tensorflownews.com/
歡迎關注我的博客,因為我也還在學習中,現有的教程經常比較大,涉及到的只是比較多,我會經常拆分出小的知識點,我的博客也會把這些小的知識點記錄下來。
推薦閱讀:
※從三大神經網路,測試對比TensorFlow、MXNet、CNTK、Theano四個框架
※TensorFlow 訓練好模型參數的保存和恢復代碼
※1.4 卷積神經網路初探
※譯文 | 與TensorFlow的第一次接觸 第六章:並發
TAG:TensorFlow | 人工智能 | 聊天机器人 |