DUMB 聊天机器人全称 Dumb Undereducated Maladroit Bot. 是使用 PyTorch 实现并使用康奈尔电影台词库训练的聊天机器人.
- 断点续训功能, 做到间断性训练.
- Migrate to Python3.
- Python 3.7
- PyTorch 1.3.1
- torchvision 0.4.2
- festival (Linux Environment)
- say (macOS Environment)
python preprocess.py
这个脚本会在./data
目录创建dialogue_corpus.txt
.
python train.py
调参可以在config.json
里面进行.
用我自己的电脑(GTX970M)训练的话, 大概需要四个半小时. 使用CPU训练请至少准备一个星期时间.
python chatbot.py
> hi .
bot: hi .
> what is your name ?
bot: vector frankenstein .
> how are you ?
bot: fine .
> where are you from ?
bot: up north .
> are you happy today ?
bot: yes .
虽然能回答一些简单的问题, 但还是特别蠢.