from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer
chatbot = ChatBot("my chat")
trainer = ListTrainer(chatbot)
注:需要将仓库克隆下来,main.py在那里,且目录与他描述的位置不同,看下报错或源码即可。
载入语料
with open('xiaohuangji.tsv', encoding='utf-8') as f:
data = f.read().replace('\t', '\n')
print(data[:100])
训练
trainer.train(data)
最后,等待训练完成之后,我们再把生成的 db.sqlite3 文件下载到本地。
注:文件目录需放置正确,同一级即可
微信公众号号聊天机器人,欢迎星标,欢迎提交pr,其中的一些原理其实我也不是很懂,所以会用就拿去用吧!不会的话,那就另寻高明!
后文:踩坑记录
chatterbot,chatterbot-corpus使用pip安装时会出现依赖包Pyyaml所需版本不兼容的情况,这时候我真的想吐槽一下,明明是两个相互配合使用的,竟然连兼容性都有些不匹配。在issue发现了问题的解决办法,但是不一定适用于所有人。
pipenv install git+https://github.com/gunthercox/chatterbot-corpus.git#egg=chatterbot-corpus
从源码进行安装
顺便记一下pipenv的简单使用方法:
shell
将在激活virtualenv的情况下生成一个shell。run
将从virtualenv运行给定命令,并转发任何参数(例如$ pipenv run python
)。check
声称当前环境已满足PEP 508要求。graph
将打印出所有已安装依赖项的漂亮图形。