一种实体关系预测方法、装置、设备及介质

本申请公开了一种实体关系预测方法、装置、设备及介质，涉及人工智能技术领域。通过输入的文本序列生成提及矩阵，进而生成第一实体关系矩阵；根据文本序列生成头实体序列和尾实体序列，进而生成第二实体关系矩阵；获取文本序列中实体整体的词嵌入表示，依此生成第三实体关系矩阵；根据第一实体关系矩阵、第二实体关系矩阵和第三实体关系矩阵进行特征整合，以预测文本序列中各实体之间的关系。由此可知，上述方案分别从矩阵特征提取、提及和实体特征提取和实体相关的文本特征提取三个角度出发，通过提取并融合上述三种特征，避免了从单一角度对实体关系的预测，能够准确预测跨句子分布实体之间的关系，提高了实体关系的预测能力。