一种主题增强的聊天机器人回复生成方法及装置

摘要

本发明提供了一种主题增强的聊天机器人回复生成方法及装置，其中的方法首先对对话文本进行预处理和采样，得到对话训练集；然后利用CNN的对对话文本就进行主题抽取，得到文本主题；再利用文本主题，通过MLP预测回复主题，得到预测回复主题；最后，将对话训练集输入到Seq2Seq生成模型，文本主题和预测回复主题通过Gate引入生成模型辅助生成过程，得到聊天机器人对话回复。本发明可以达到降低高频一般回复的生成概率，使神经对话模型能生成更加连贯、丰富的回复的技术效果。