ChatGirl 一個基於 TensorFlow Seq2Seq 模型的聊天機器人

簡介

還在開發中,它工作的效果還不好。但是你可以直接訓練,並且運行。

包含預處理過的 twitter 英文數據集,訓練,運行,工具代碼,可以運行但是效果有待提高。

數據集

Twitter 數據集:

github.com/suriyadeepan

訓練

你需要新建一個 model 文件夾來保存訓練完的模型

運行這個文件來訓練模型

Train_Model.py

運行

運行這個文件來運行模型

Run_model.py

工具集

w2idx:單詞到 id ,用於下一步將單詞轉化為向量。

idx2w:id 到 單詞,用於將 id 轉換為單詞。

hello.py 中有兩個函數的 demo。

結果

Result.md(部分輸出)

博客

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 中文入門實戰項目)

github.com/fendouai/Fac

  • TensorFlow LSTM Model Project:

一個比特幣交易機器人基於 Tensorflow LSTM 模型,僅供娛樂。 A Bitcoin trade robot based on Tensorflow LSTM model.Just for fun.

github.com/TensorFlowNe

  • TensorFlow Seq2Seq Model Project:

ChatGirl is an AI ChatBot based on TensorFlow Seq2Seq Model.ChatGirl 一個基於 TensorFlow Seq2Seq 模型的聊天機器人。(包含預處理過的 twitter 英文數據集,訓練,運行,工具代碼,可以運行但是效果有待提高。)

github.com/fendouai/Cha

學習流程

如果看到這裡有很多不懂的話,建議:

* Hello World

zhuanlan.zhihu.com/p/27

* 基本概念

zhuanlan.zhihu.com/p/27

* 卷積神經網路

zhuanlan.zhihu.com/p/28

* 訓練好模型參數的保存和恢復代碼

zhuanlan.zhihu.com/p/27

* TensorFlowNews 專欄

zhuanlan.zhihu.com/Tens

* TensorFlowNews 博客

tensorflownews.com/

歡迎關注我的博客,因為我也還在學習中,現有的教程經常比較大,涉及到的只是比較多,我會經常拆分出小的知識點,我的博客也會把這些小的知識點記錄下來。

推薦閱讀:

從三大神經網路,測試對比TensorFlow、MXNet、CNTK、Theano四個框架
TensorFlow 訓練好模型參數的保存和恢復代碼
1.4 卷積神經網路初探
譯文 | 與TensorFlow的第一次接觸 第六章:並發

TAG:TensorFlow | 人工智能 | 聊天机器人 |