实体关系抽取模型的训练方法、实体关系抽取方法及设备
Abstract:
本申请公开了一种实体关系抽取模型的训练方法、实体关系抽取方法及设备,方法包括在预设关系类型集中选取一个目标关系类型,通过生成模型生成若干个符合所述目标关系类型的句子‑三元组对以构建合成数据集,通过初始抽取模型确定所述合成数据集中的每个句子‑三元组对的奖励函数值;基于所有奖励函数值对生成模型和所述初始抽取模型进行训练,以得到抽取模型。本申请实施例基于奖励函数值对生成模型和抽取模型进行训练,使得生成模型和抽取模型相互促进,增强了抽取模型在未知关系类型上的泛化能力,同时也提高了生成模型确定合成数据的数据质量,从而可以提高训练得到的抽取模型的准确性,进而可以提高基于抽取模型确定实体关系的准确性。
Patent Agency Ranking
0/0