一种模型训练方法、命名实体识别方法、介质及设备

    公开(公告)号:CN117952109A

    公开(公告)日:2024-04-30

    申请号:CN202410105933.0

    申请日:2024-01-24

    Abstract: 本发明公开了一种模型训练方法、命名实体识别方法、介质及设备,涉及水文地质命名实体识别领域,所述模型训练方法,主要包括以下步骤:获取生成式PLM,获取源序列、目标序列,将源序列送入编码器,将目标序列依次送入解码器;获取解码器输出的第c个词的条件概率;获取损失函数,调整模型参数比较不同模型参数下损失函数的值,在损失函数收敛时确定模型参数,得到训练模型。实施本发明提供的模型训练方法、命名实体识别方法、介质及设备,可以克服当前的命名实体识别方法在水文地质领域效率低、精度差等缺陷。

Patent Agency Ranking