[发明专利]对话推荐模型的训练方法、装置、设备和存储介质

权利要求书

1.一种对话推荐模型的训练方法，包括：

基于两个机器学习模型之间的对话信息，获得目标领域的主动引导语料，所述主动引导语料中包括所述目标领域的推荐目标；

基于所述目标领域的知识图谱，获得所述推荐目标对应的知识对话语料；

基于所述主动引导语料和所述知识对话语料，获得所述目标领域的对话推荐语料；

采用所述目标领域的对话推荐语料，训练所述目标领域的对话推荐模型。

2.根据权利要求1所述的方法，其中，所述两个机器学习模型包括：主动引导模型和开放式对话模型，所述主动引导模型的输入包括所述目标领域对应的关键词，所述基于两个机器学习模型之间的对话信息，获得目标领域的主动引导语料，包括：

采用所述主动引导模型和所述开放式对话模型，进行多轮对话，直至获得特定对话信息，所述特定对话信息为包括所述目标领域的推荐目标的所述主动引导模型输出的对话信息，所述主动引导模型输出的对话信息基于所述关键词获得；

将所述特定对话信息之前的对话信息，作为所述目标领域的主动引导语句，所述对话信息包括：所述开放式对话模型输出的对话信息和所述主动引导模型输出的对话信息。

3.根据权利要求1所述的方法，其中，所述基于所述目标领域的知识图谱，获得所述推荐目标对应的知识对话语料，包括：

基于所述目标领域的知识图谱，将所述推荐目标转换为问答对，所述问答对为多对；

对所述多对的问答对进行排列组合，和/或，对所述多对的问答对与挖掘的所述目标领域的对话语料进行排列组合，以获得所述推荐目标对应的知识对话语料。

4.根据权利要求1所述的方法，其中，所述基于所述主动引导语料和所述知识对话语料，获得所述目标领域的对话推荐语料，包括：

将所述主动引导语料和所述知识对话语料，组合成所述目标领域的对话推荐语料。

5.根据权利要求1-4任一项所述的方法，其中，所述采用所述目标领域的对话推荐语料，训练所述目标领域的对话推荐模型，包括：

采用所述目标领域的对话推荐语料，对已有的预训练模型进行训练，以获得所述目标领域的对话推荐模型。

6.一种对话推荐模型的训练装置，包括：

主动引导模块，用于基于两个机器学习模型之间的对话信息，获得目标领域的主动引导语料，所述主动引导语料中包括所述目标领域的推荐目标；

知识对话模块，用于基于所述目标领域的知识图谱，获得所述推荐目标对应的知识对话语料；

语料构造模块，用于基于所述主动引导语料和所述知识对话语料，获得所述目标领域的对话推荐语料；

训练模块，用于采用所述目标领域的对话推荐语料，训练所述目标领域的对话推荐模型。

7.根据权利要求6所述的装置，其中，所述两个机器学习模型包括：主动引导模型和开放式对话模型，所述主动引导模型的输入包括所述目标领域对应的关键词，所述主动引导模块具体用于：

采用所述主动引导模型和所述开放式对话模型，进行多轮对话，直至获得特定对话信息，所述特定对话信息为包括所述目标领域的推荐目标的所述主动引导模型输出的对话信息，所述主动引导模型输出的对话信息基于所述关键词获得；

将所述特定对话信息之前的对话信息，作为所述目标领域的主动引导语句，所述对话信息包括：所述开放式对话模型输出的对话信息和所述主动引导模型输出的对话信息。

8.根据权利要求6所述的装置，其中，所述知识对话模块具体用于：

基于所述目标领域的知识图谱，将所述推荐目标转换为问答对，所述问答对为多对；

对所述多对的问答对进行排列组合，和/或，对所述多对的问答对与挖掘的所述目标领域的对话语料进行排列组合，以获得所述推荐目标对应的知识对话语料。

9.根据权利要求6所述的装置，其中，所述语料构造模块具体用于：

将所述主动引导语料和所述知识对话语料，组合成所述目标领域的对话推荐语料。