使用 ChatterBot 来构建聊天机器人

当你在搜索引擎中输入“如何建立聊天机器人?” 时,你可能会发现 ChatterBot 这款基于机器学习(machine learning)、对话引擎(conversational dialog engine)来创建聊天的机器人(chat bots)。

ChatterBot 的工作方式

processdiagram

Image source: ChatterBot

在本文中,我们将介绍如何仅在5分钟内使用ChatterBot构建聊天机器人。

准备

首先下载 ChatterBot

pip install ChatterBot

创建文件并命名为 chat.py

#import ChatBot
from chatterbot import ChatBot

传经一个新的聊天机器人并为其赋予一个名字(我使用的名字是 “Candice”):

bot = ChatBot('Candice')

您的机器人已创建,但此时您的机器人尚不了解,因为您必须对它进行基于一定数据的训练。

PS:默认情况下,ChatterBot 库将创建一个 Sqlite 数据库来建立一些聊天语句。

对机器人进行数据训练

#import ListTrainer
from chatterbot.trainers import ListTrainer

bot.set_trainer(ListTrainer)
# Training 
bot.train(['What is your name?', 'My name is Candice'])
bot.train(['Who are you?', 'I am a bot, created by you' ])

您的机器人现已接受 2 条陈述的训练。当您问您的机器人 What is your name? 时,它会以“我的名字叫 My name is Candice 答复。

您还可以对多个语句进行训练,例如:

bot.train(['Do you know me?', 'Yes, you created me', 'No', 'Sahil?', 'No idea'])

如您所见,很难在每个语句上训练机器人。因此,我们将使用 ChatterBotCorpusTrainer 在大型数据集上训练我们的机器人:

from chatterbot.trainers import ChatterBotCorpusTrainer
# Create a new trainer for the chatbot
trainer = ChatterBotCorpusTrainer(bot)

# Train the chatbot based on the english corpus
trainer.train("chatterbot.corpus.english")

# Get a response to an input statement
chatbot.get_response("Hello, how are you today?")

或者您可以下载不同语言的数据集。并通过您选择的语言对其进行训练。

假设我们加载完成英语模式的训练集后,我们可以像这样训练我们的机器人:

for files in os.listdir('./english/'):
    data=open('./english/'+files,'r').readlines()
    bot.train(data)

注意:确保数据集和程序文件在同一文件夹中,否则请编辑路径。

聊天特征

# To exit say "Bye"
while True:
        # Input from user
    message=input('\t\t\tYou:')
        #if message is not "Bye"
    if message.strip()!='Bye':
        reply=bot.get_response(message)
        print('Candice:',reply)
        # if message is "Bye"
    if message.strip()=='Bye':
        print('Candice: Bye')
        break

让我们尝试在终端跑一下代码:

python chat.py

他会先对机器人进行训练,然后您可以和其进行聊天。

源码

#import libraries
from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer
import os

#Create a chatbot
bot=ChatBot('Candice')
bot.set_trainer(ListTrainer)

#training on english dataset
for files in os.listdir('./english/'):
    data=open('./english/'+files,'r').readlines()
    bot.train(data)

#chat feature
while True:
    message=input('\t\t\tYou:')
    if message.strip()!='Bye':
        reply=bot.get_response(message)
        print('Candice:',reply)
    if message.strip()=='Bye':
        print('Candice: Bye')
        break
本文中的所有译文仅用于学习和交流目的,转载请务必注明文章译者、出处、和本文链接
我们的翻译工作遵照 CC 协议,如果我们的工作有侵犯到您的权益,请及时联系我们。

原文地址:https://dev.to/sahilrajput/build-your-fi...

译文地址:https://learnku.com/python/t/38806

本文为协同翻译文章,如您发现瑕疵请点击「改进」按钮提交优化建议
讨论数量: 2

温馨提示:要做微信机器人还需要配合一下itchat

3年前 评论

讨论应以学习和精进为目的。请勿发布不友善或者负能量的内容,与人为善,比聪明更重要!