模型的训练方法、装置以及计算机可读存储介质

    公开(公告)号:CN115828923A

    公开(公告)日:2023-03-21

    申请号:CN202211443788.4

    申请日:2022-11-17

    Abstract: 本发明公开了一种模型的训练方法、装置以及计算机可读存储介质,该方法包括:获取当前批次的第一样本数据;将第一样本数据输入目标模型,得到每个样本话语的第一预测情感标签;屏蔽第一样本数据中的上下文表征向量或者外部知识表征向量,得到第二样本数据;将第二样本数据输入目标模型,得到每个样本话语的第二预测情感标签;针对每个样本话语,根据样本话语的第一预测情感标签与第二预测情感标签是否相同,设置对应的第一目标标签;根据每个样本话语的话语表征向量以及每个样本话语的第一目标标签进行有监督对比损失计算;根据得到的第一损失值,对目标模型进行训练。本申请的方法能够提升模型的性能。

Patent Agency Ranking