一种实体关系识别方法、装置、设备及存储介质

    公开(公告)号:CN115293149A

    公开(公告)日:2022-11-04

    申请号:CN202210936756.1

    申请日:2022-08-05

    摘要: 本发明公开了一种实体关系识别方法、装置、设备及存储介质。该方法包括:通过提示学习模块,将输入的待识别文本数据转换为提示文本数据后,输入实体抽取模型输出数据实体集合和编码特征向量;实体抽取模型基于提示数据训练样本集迭代训练预训练实体抽取模型得到;通过图神经网络模块,基于数据实体集合和编码特征向量构建实体关系图,对实体关系图中的各节点进行分类得到实体类别;通过实体关系识别模块,对各实体类别进行关系分类输出所述待识别文本数据的实体关系,能够在少量标注数据样本的条件下得到较好的实体关系识别效果,减少实体抽取模型的训练对训练样本的数量需求、降低文本数据训练样本的标记成本,减少计算量和对设备的硬件要求。