[发明专利]文本分类模型训练方法和装置在审
申请号: | 202010158022.6 | 申请日: | 2020-03-09 |
公开(公告)号: | CN111552802A | 公开(公告)日: | 2020-08-18 |
发明(设计)人: | 李宁 | 申请(专利权)人: | 北京达佳互联信息技术有限公司 |
主分类号: | G06F16/35 | 分类号: | G06F16/35;G06F40/30;G06K9/62;G06N3/04 |
代理公司: | 北京先进知识产权代理有限公司 11648 | 代理人: | 叶碧莲;赵霞兵 |
地址: | 100085 北京市海淀*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 文本 分类 模型 训练 方法 装置 | ||
1.一种文本分类模型训练方法,其特征在于,包括:
获取样本文本;单个所述样本文本包括:正文和与正文对应的标题;
将所述样本文本输入待训练的文本分类模型,以生成所述样本文本的正文在不同维度下的正文语义特征向量,和所述样本文本的标题在所述不同维度下的标题语义特征向量;其中,所述不同维度中的最大维度,基于预设的文本分类标签的数量确定;
从所述不同维度下的正文语义特征向量和所述不同维度下的标题语义特征向量中,选取至少一对匹配于相同的所述文本分类标签、且具备相同维度的正文语义特征向量和标题语义特征向量,作为第一类语义特征向量对;
基于所述样本文本的正负样本类型,采用调整所述文本分类模型的参数,以使得第一类语义特征向量对中的正文语义特征向量的值和所述第一类语义特征向量对中的标题语义特征向量的值最大化或最小化的方式,执行对所述待训练的文本分类模型的训练。
2.根据权利要求1所述的文本分类模型训练方法,其特征在于,
基于所述样本文本的正负样本类型,采用调整所述文本分类模型的参数,以使得第一类语义特征向量对中的正文语义特征向量的值和所述第一类语义特征向量对中的标题语义特征向量的值最大化或最小化的方式,执行对所述待训练的文本分类模型的训练,具体包括:
若所述样本文本的类型为正样本,则采用调整所述文本分类模型的参数,以使得所述第一类语义特征向量对中的正文语义特征向量的值和所述第一类语义特征向量对中的标题语义特征向量的值最大化的方式,执行所述文本分类模型的训练;
若所述样本文本的类型为负样本,则采用调整所述文本分类模型的参数,以使得所述第一类语义特征向量对中的正文语义特征向量的值和所述第一类语义特征向量对中的标题语义特征向量的值最小化的方式,执行所述文本分类模型的训练。
3.根据权利要求2所述的文本分类模型训练方法,其特征在于,所述方法还包括:
从所述不同维度下的正文语义特征向量和所述不同维度下的标题语义特征向量中,选取至少一对匹配于不同的所述文本分类标签、且具备相同维度的正文语义特征向量和标题语义特征向量,作为第二类语义特征向量对;则,
所述基于所述样本文本的正负样本类型,采用调整所述文本分类模型的参数,以使得第一类语义特征向量对中的正文语义特征向量的值和所述第一类语义特征向量对中的标题语义特征向量的值最大化或最小化的方式,执行对所述待训练的文本分类模型的训练,还包括:
若所述样本文本的类型为正样本,则采用调整所述文本分类模型的参数,以使得所述第二类语义特征向量对中的正文语义特征向量的值和所述第二类语义特征向量对中的标题语义特征向量的值最小化的方式,执行所述文本分类模型的训练。
4.根据权利要求3所述的文本分类模型训练方法,其特征在于,从所述不同维度下的正文语义特征向量和所述不同维度下的标题语义特征向量中,选取至少一对匹配于相同的所述文本分类标签、且具备相同维度的正文语义特征向量和标题语义特征向量,作为第一类语义特征向量对,具体包括:
按照所述不同维度下的正文语义特征向量的值由大至小的顺序,依次选取指定数量的正文语义特征向量;
按照所述不同维度下的标题语义特征向量的值的由大至小的顺序,依次选取所述指定数量的标题语义特征向量;
从选取的所述指定数量的正文语义特征向量和所述指定数量的标题语义特征向量中,选取至少一对匹配于相同所述文本分类标签、且具备相同维度的正文语义特征向量和标题语义特征向量,作为第一类语义特征向量对。
5.根据权利要求1~4任一权项所述的文本分类模型训练方法,其特征在于,所述待训练的文本分类模型包括伪孪生神经网络;则,
将所述样本文本输入待训练的文本分类模型,以生成所述样本文本的正文在不同维度下的正文语义特征向量,和所述样本文本的标题在所述不同维度下的标题语义特征向量,具体包括:
将所述样本文本中的正文和标题分别输入到伪孪生神经网络的两侧神经网络中,以得到所述样本文本的正文在不同维度下的正文语义特征向量,以及所述样本文本的标题在所述不同维度下的标题语义特征向量。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京达佳互联信息技术有限公司,未经北京达佳互联信息技术有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010158022.6/1.html,转载请声明来源钻瓜专利网。