嘿,我正在尝试使用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)
感谢您查看我的问题。
答案 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
使用此功能,您只需将参数传递到文件的路径,它将返回该文件中的行列表。