如何解决ChatterBot仅返回多行答案第一行的问题?
问题:ChatterBot ListTrainer 返回多行答案时仅输出第一行
嘿,我来帮你解决这个问题!这其实是文件读取逻辑和ListTrainer的工作方式不匹配导致的,咱们一步步理清楚并搞定它。
你的代码
from chatterbot.trainers import ListTrainer from chatterbot import ChatBot bot = ChatBot('MyBot') conv = open('chats.txt','r').readlines() bot.set_trainer(ListTrainer) bot.train(conv) while True: request = input('You:') response = bot.get_response(request) print('Bot:',response)
chats.txt 示例
男性压力或抑郁的常见预警信号有哪些? 压力的一般表现: 心率加快 肌肉紧张 血压升高 胃部紧绷 长期压力的表现: 频繁感冒或流感 头痛 睡眠障碍 皮肤问题
问题根源
当你用readlines()读取文件时,每一行都会被拆成列表conv里的独立元素。而ListTrainer的训练逻辑是把列表元素按顺序当成对话轮次:第1个元素是用户提问,第2个是机器人回复,第3个是下一个提问,第4个是下一个回复……
所以你的chats.txt里,问题是第一个元素,接下来的每一行答案都被当成了单独的对话元素——机器人匹配到问题后,只会返回紧随其后的第一个元素(也就是“压力的一般表现:”),后面的行被错误地当成了下一轮的用户输入,自然不会被返回。
解决方案
我们需要把连续的多行答案合并成一个完整的字符串,作为对应问题的回复。这里提供两种实用方法:
方法1:修改chats.txt格式(简单直接)
把多行答案合并成一行,用\n换行符分隔,这样读取后一个元素就是完整的答案:
男性压力或抑郁的常见预警信号有哪些? 压力的一般表现:\n心率加快\n肌肉紧张\n血压升高\n胃部紧绷\n长期压力的表现:\n频繁感冒或流感\n头痛\n睡眠障碍\n皮肤问题
然后你的原代码只需微调,去掉每行末尾的多余换行符:
from chatterbot.trainers import ListTrainer from chatterbot import ChatBot bot = ChatBot('MyBot') # 读取并处理每行,去掉末尾换行符 conv = [line.rstrip('\n') for line in open('chats.txt','r')] bot.set_trainer(ListTrainer) bot.train(conv) while True: request = input('You: ') response = bot.get_response(request) print('Bot:', response)
方法2:不修改文件,代码中自动合并多行答案(更灵活)
如果不想改动chats.txt的格式,可以在读取时自动识别问题行(比如以中文问号“?”结尾),然后把后续的非问题行合并成一个完整的答案:
from chatterbot.trainers import ListTrainer from chatterbot import ChatBot bot = ChatBot('MyBot') # 读取文件,过滤空行并去掉每行首尾空白 with open('chats.txt', 'r', encoding='utf-8') as f: lines = [line.strip() for line in f if line.strip()] conv = [] i = 0 while i < len(lines): # 添加问题行 conv.append(lines[i]) i += 1 # 收集后续所有非问题行作为答案 answer_lines = [] # 可根据实际情况调整问题行的判断规则(比如英文用"?") while i < len(lines) and not lines[i].endswith('?'): answer_lines.append(lines[i]) i += 1 # 用换行符连接所有答案行,作为完整回复 conv.append('\n'.join(answer_lines)) bot.set_trainer(ListTrainer) bot.train(conv) while True: request = input('You: ') response = bot.get_response(request) print('Bot:', response)
效果测试
运行修改后的代码,当你输入问题“男性压力或抑郁的常见预警信号有哪些?”时,机器人会返回完整的多行答案:
Bot: 压力的一般表现: 心率加快 肌肉紧张 血压升高 胃部紧绷 长期压力的表现: 频繁感冒或流感 头痛 睡眠障碍 皮肤问题
内容的提问来源于stack exchange,提问作者AnalyticsPy
相关产品推荐
相关产品推荐

