文本分类模型的训练方法、文本分类方法及相关设备
摘要:
本申请提供了一种文本分类模型的训练方法、装置及电子设备,用于提升文本分类模型的训练效率。该方法包括:根据原始类别样本以及新增类别样本生成样本训练集;通过预训练分类模型中的第一特征提取模块,得到类别无关特征,以及通过预训练分类模型中的第二特征提取模块,得到类别相关特征;通过类别无关任务模块的输出以及预设的类别无关损失函数,对预训练分类模型中的第一特征提取模块进行更新;通过类别相关任务模块的输出以及预设的类别相关损失函数,对预训练分类模型中的第二特征提取模块进行更新;根据更新后的第一特征提取模块以及更新后的第二特征提取模块,对预训练分类模型中包含的合并分类模块进行更新,以生成文本分类模型。
0/0