文本搜索模型的训练方法及装置、计算设备与流程

文档序号:34253395发布日期:2023-05-25 02:57阅读:来源:国知局

技术特征:

1.一种文本搜索模型的训练方法,其特征在于,所述文本搜索模型包括编码器,所述方法包括:

2.根据权利要求1所述的文本搜索模型的训练方法,其特征在于,所述n条样本事件文本所属的事件类别各不相同。

3.根据权利要求1所述的文本搜索模型的训练方法,其特征在于,获取当前批次的训练数据包括:

4.根据权利要求3所述的文本搜索模型的训练方法,其特征在于,所述标签矩阵中第j列的元素中仅单个元素为1,所述规范文本数据库包括多个规范文本集合,根据所述标签矩阵中第j列的n个元素和所述n条样本事件文本,从规范文本数据库中抽取所述第j条样本规范文本包括:

5.根据权利要求4所述的文本搜索模型的训练方法,其特征在于,所述规范文本集合包括多个样本规范文本单元,从所述第j列的正例事件文本关联的规范文本集合中抽取所述第j条样本规范文本包括:

6.根据权利要求1所述的文本搜索模型的训练方法,其特征在于,所述标签矩阵中第i行第i列的元素均为1,其余元素均为0。

7.根据权利要求1所述的文本搜索模型的训练方法,其特征在于,所述编码器包括第一编码器和第二编码器,采用所述编码器分别对所述样本事件文本组和所述样本规范文本组进行编码包括:

8.根据权利要求7所述的文本搜索模型的训练方法,其特征在于,所述第一编码器和所述第二编码器均为句向量编码器。

9.根据权利要求1所述的文本搜索模型的训练方法,其特征在于,所述方法还包括:

10.一种文本搜索模型的训练装置,其特征在于,所述文本搜索模型包括编码器,所述装置包括:

11.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器运行时,执行权利要求1至9中任一项所述的文本搜索模型的训练方法的步骤。

12.一种计算设备,包括存储器和处理器,所述存储器上存储有可在所述处理器上运行的计算机程序,其特征在于,所述处理器运行所述计算机程序时执行权利要求1至9中任一项所述的文本搜索模型的训练方法的步骤。


技术总结
一种文本搜索模型的训练方法及装置、计算设备,方法包括:获取当前批次的训练数据,当前批次的训练数据包括样本事件文本组和样本规范文本组,样本事件文本组包括N条样本事件文本,样本规范文本组包括N条样本规范文本;根据预测矩阵和标签矩阵,计算模型损失值,其中,预测矩阵中第i行第j列的元素是根据样本事件文本组中第i条样本事件文本的向量表示和样本规范文本组中第j条样本规范文本的向量表示计算得到的相似度,标签矩阵中第i行第j列的元素用于指示第i条样本事件文本和第j条样本规范文本是否匹配;根据模型损失值更新文本搜索模型中的编码器。通过本申请实施例提供的训练方法,能够训练得到准确度较高的文本搜索模型。

技术研发人员:姚昱材,张晓丹
受保护的技术使用者:华院计算技术(上海)股份有限公司
技术研发日:
技术公布日:2024/1/12
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1