可以使用ChatBot添加对话语句词汇训练生成新的聊天机器人,代码如下:
#!/usr/local/python3/bin/python3
# -*- coding: utf-8 -*
from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer
from chatterbot.trainers import ChatterBotCorpusTrainer
# creating a chatbot
myBot = ChatBot(
name = 'Sakura',
read_only = True,
logic_adapters = [
'chatterbot.logic.MathematicalEvaluation',
'chatterbot.logic.BestMatch'
]
)
# training the chatbot
small_convo = [
'在吗!',
'在',
'最近怎么样?',
'还好啊,你呢?',
'我也还好.',
'吃饭没.',
'吃了',
'今天的午饭不错',
'吃了啥',
]
math_convo_1 = [
'亚运会',
'亚洲运动会.'
]
math_convo_2 = [
'Law of Cosines',
'c**2 = a**2 + b**2 - 2*a*b*cos(gamma)'
]
# using the ListTrainer class
list_trainee = ListTrainer(myBot)
for i in (small_convo, math_convo_1, math_convo_2):
list_trainee.train(i)
# using the ChatterBotCorpusTrainer class
corpus_trainee = ChatterBotCorpusTrainer(myBot)
corpus_trainee.train('chatterbot.corpus.chinese')
while True:
message = input("\n")
print('您的输入:{}'.format(message))
print("机器的回复:{}".format(myBot.get_response(message)))
网友回复