斯坦福的聊天机器人代码,亲测可运行。
https://github.com/chiphuyen/stanford-tensorflow-tutorials/tree/master/2017/assignments/chatbot
Step 1: create a data folder in your project directory, download the Cornell Movie-Dialogs Corpus fromhttps://www.cs.cornell.edu/~cristian/Cornell_Movie-Dialogs_Corpus.html Unzip it
Step 2: python data.py
This will do all the pre-processing for the Cornell dataset.
Step 3: python chatbot.py --mode [train/chat]
If mode is train, then you train the chatbot. By default, the model will restore the previously trained weights (if there is any) and continue training up on that.
If you want to start training from scratch, please delete all the checkpoints in the checkpoints folder.
If the mode is chat, you'll go into the interaction mode with the bot.
By default, all the conversations you have with the chatbot will be written into the file output_convo.txt in the processed folder.
其他资料:
https://github.com/warmheartli/ChatBotCourse
这是一个动手做聊天机器人教程。已经更新了5个版本了。