=================================================更新================================================== ========= 訓練好的模型已經上傳到百度雲網盤,如果大家有需要可以去下載。模型訓練速度的話,CPU,16G內存,一天即刻訓練完成~~~
連結:https://pan.baidu.com/s/1hrNxaSk 密碼:d2sn
=================================================分割線,下面是文字=============================================== ==
本文是一個簡單的基於seq2seq模型的chatbot對話系統的tensorflow實作。
程式碼的講解可以參考我的知乎專欄文章:
從頭實現深度學習的對話系統--簡單chatbot程式碼實現
程式碼參考了DeepQA,在其基礎上加入了beam search的功能和attention的機制,
最終的效果如下圖所示:
測試效果,依照使用者輸入回覆機率最大的前beam_size個句子:
#使用方法
1,下載程式碼到本地(data資料夾下已經包含了處理好的資料集,所以無需額外下載資料集)
2,訓練模型,將chatbot.py檔案第34行的decode參數修改為False,進行訓練模型
(之後我會把我這裡訓練好的模型上傳到網路上方便大家使用)
3,訓練完後(大概一天左右的時間,30個epoches),再將decode參數修改為True
就可以進行測試了。輸入你想問的話看他回什麼==
這裡還要注意的就是要記得修改資料集和最後模型檔的絕對路徑,不然可能會報錯。
分別在44行,57行,82行三處。好了,接下來就可以愉快的玩耍了~~