chatterbot:我的chatbot无法接受训练

时间:2018-06-26 08:05:37

标签: python python-3.x chatbot chatterbot

嘿,我正在尝试使用chatterbot用现有的主体训练我的chatbot,但是我发现我的chatbot无法训练。似乎在bot.train(data)行中被卡住了。这是我的代码:

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

bot = ChatBot('ChatBot')
bot.set_trainer(ListTrainer)
new = []

for files in os.listdir('/Users/christinalai/Desktop/corpus'):
    data = open('/Users/christinalai/Desktop/corpus/' + files, mode='r',errors="ignore").readlines()
    for line in data:
        new.append(line)

bot.train(new)

while True:
    message = input("You: ")
    if message.strip() not in message_list:
        reply = bot.get_response(message)
        print("ChatBot: ", reply)

感谢您查看我的问题。

1 个答案:

答案 0 :(得分:0)

您可以使用此功能以列表形式检索您的copus内容。然后您可以照常训练聊天机器人。

def delcha(path):
    with open(text_file,encoding='utf-8') as file:
        list_line_file = file.readlines()
    for ans in list_line_file:
        if '\n' in ans:
            edit_ans = ans[0:len(ans)-1]
            index = list_line_file.index(ans)
            list_line_file.pop(index)
            list_line_file.insert(index, edit_ans)
    return list_line_file

使用此功能,您只需将参数传递到文件的路径,它将返回该文件中的行列表。