一种模型训练方法、数据处理方法、装置及电子设备

基本信息

申请号 CN202110227463.1 申请日 -
公开(公告)号 CN112926737A 公开(公告)日 2021-06-08
申请公布号 CN112926737A 申请公布日 2021-06-08
分类号 G06N3/08;G06N3/04;G06F16/951;G06F40/30;G06K9/62 分类 计算;推算;计数;
发明人 张发恩;刘雨微 申请(专利权)人 创新奇智(上海)科技有限公司
代理机构 北京超凡宏宇专利代理事务所(特殊普通合伙) 代理人 余菲
地址 201900 上海市宝山区上大路668号1幢569N
法律状态 -

摘要

摘要 本申请涉及一种模型训练方法、数据处理方法、装置及电子设备,属于计算机技术领域。该训练方法包括:获取数据样本集,并按照预设比例划分为训练集和测试集,其中,所述数据样本集包括:网页中与指定目标相关的多条正面评论文本和多条负面评论文本;利用所述训练集和所述测试集对BERT模型进行迭代训练,得到训练好的情感分类模型,其中,所述BERT模型的输入层经过嵌入层后,沿着句子序列长度方向分别做平均池化和最大池化,并将两者级联。通过对BERT模型进行改进,通过将每一句话的平均响应和最大响应考虑在内,使得训练出的词向量不仅含有语义信息,而且还将整个句子的情感极性表达考虑在内,从而提高了情感分类预测的准确性。