嵌套命名实体识别模型的训练方法、装置、设备和介质
摘要:
本发明涉及嵌套命名实体识别模型的训练方法、装置、设备和介质,该方法包括:获取标注好的数据集,根据数据集确定第一数据集,根据第一数据集确定实体词边界开始向量和实体词边界结束向量;根据实体词边界开始向量和实体词边界结束向量确定命名实体识别预测得分矩阵;根据命名实体识别预测得分矩阵确定文字样本对应的实体类型的预测结果;根据第一数据集和实体类型的预测结果计算得到实体类型预测损失;根据实体类型预测损失调整模型参数得到嵌套命名实体识别模型。本申请没有使用堆叠式的模型,避免了错误传递的问题,并根据实体词边界词确定实体类型预测损失,通过预测损失调整模型参数得到嵌套命名实体识别模型,提高了嵌套命名实体识别能力。
0/0