发明公开
- 专利标题: 文本分类模型的训练方法、文本分类方法及相关设备
-
申请号: CN202310487043.6申请日: 2023-04-28
-
公开(公告)号: CN117493552A公开(公告)日: 2024-02-02
- 发明人: 陈文俊 , 蒋宁 , 陆全 , 夏粉 , 肖冰 , 李宽
- 申请人: 马上消费金融股份有限公司
- 申请人地址: 重庆市渝北区黄山大道中段52号渝兴广场B2栋4至8楼
- 专利权人: 马上消费金融股份有限公司
- 当前专利权人: 马上消费金融股份有限公司
- 当前专利权人地址: 重庆市渝北区黄山大道中段52号渝兴广场B2栋4至8楼
- 代理机构: 北京天昊联合知识产权代理有限公司
- 代理商 彭瑞欣; 冯建基
- 主分类号: G06F16/35
- IPC分类号: G06F16/35 ; G06F18/214 ; G06F18/241 ; G06F18/213 ; G06F18/23
摘要:
本申请提供了一种文本分类模型的训练方法、装置及电子设备,用于提升文本分类模型的训练效率。该方法包括:根据原始类别样本以及新增类别样本生成样本训练集;通过预训练分类模型中的第一特征提取模块,得到类别无关特征,以及通过预训练分类模型中的第二特征提取模块,得到类别相关特征;通过类别无关任务模块的输出以及预设的类别无关损失函数,对预训练分类模型中的第一特征提取模块进行更新;通过类别相关任务模块的输出以及预设的类别相关损失函数,对预训练分类模型中的第二特征提取模块进行更新;根据更新后的第一特征提取模块以及更新后的第二特征提取模块,对预训练分类模型中包含的合并分类模块进行更新,以生成文本分类模型。