[发明专利]知识图谱预测的方法和装置

权利要求书

1.一种知识图谱预测的方法，其特征在于，包括：

获取第一实体和第二实体，所述第一实体为所述知识图谱中的实体对象之一，所述第二实体为所述知识图谱中的实体对象之一；

根据所述第一实体和所述第二实体，建立第一三角子图，所述第一三角子图包括第一三元组、第二三元组和第三三元组，其中，所述第一三元组包括所述第一实体、所述第二实体和候选关系，所述第二三元组包括所述第一实体、第一共同邻居实体和所述第一实体与所述第一共同邻居实体的第一关系，所述第三三元组包括所述第二实体、所述第一共同邻居实体和所述第二实体与所述第一共同邻居实体的第二关系，所述第一共同邻居实体为所述知识图谱中的实体对象之一，所述第一关系为所述知识图谱中的关系之一，所述第二关系为所述知识图谱中的关系之一，所述候选关系为所述知识图谱中的关系之一；

根据所述第一三角子图和嵌入式模型，确定所述第一实体和所述第二实体之间的关系，所述嵌入式模型包括所述知识图谱中的已知的所有实体对象和所述所有实体对象之间已知的关系；

根据所述第一实体和所述第二实体之间的关系，更新所述知识图谱，所述更新后的知识图谱中包括所述第一实体和所述第二实体之间的关系，所述更新后的知识图谱用于基于用户提供的查询词搜索与所述查询词相关的信息；

所述根据所述第一三角子图和嵌入式模型，确定所述第一实体和所述第二实体之间的关系，包括：

根据所述嵌入式模型的第一损失函数，确定所述第一三角子图的第一损失值，所述第一损失函数为三角子图的能量与损失值的映射关系，所述第一三角子图的能量包括所述第一三角子图包括的三个三元组的能量与关系能量的矢量和，其中，所述三个三元组中的每个三元组的能量包括所述每个三元组中的左实体的向量与所述每个三元组中的关系向量之和与所述每个三元组中的右实体的向量之间的距离，所述关系能量包括所述第一三元组中的关系的向量与所述第二三元组中的关系的向量之和与所述第三三元组中的关系的向量之间的距离；

根据所述第一三角子图的第一损失值，确定所述第一实体和所述第二实体之间的关系；

所述根据所述嵌入式模型的第一损失函数，确定所述第一三角子图的第一损失值，包括：

确定所述第一三角子图的能量；

根据第一三角子图的能量和所述第一损失函数，确定所述第一损失值。

2.根据权利要求1所述的方法，其特征在于，所述根据所述第一实体和所述第二实体，建立第一三角子图之前，所述方法还包括：

从所述知识图谱中获取所述候选关系；

所述根据所述第一三角子图的第一损失值，确定所述第一实体和所述第二实体之间的关系，包括：

若所述第一损失值小于或等于预设值，将所述候选关系作为所述第一实体和所述第二实体之间的关系。

3.根据权利要求1所述的方法，其特征在于，所述根据所述第一实体和所述第二实体，建立第一三角子图之前，所述方法还包括：

从所述知识图谱中获取至少两个关系；

将所述至少两个关系中的每个关系作为所述候选关系，分别建立与所述至少两个关系一一对应的至少两个第一三角子图；

所述根据所述第一三角子图和嵌入式模型，确定所述第一实体和所述第二实体之间的关系，包括：

确定所述至少两个第一三角子图中的每个第一三角子图的第一损失值；

从所述至少两个第一三角子图的第一损失值中确定最小第一损失值；

将所述最小第一损失值对应的第一三角子图的候选关系作为所述第一实体与所述第二实体之间的关系。

4.根据权利要求1所述的方法，其特征在于，所述根据所述第一三角子图和嵌入式模型，确定所述第一实体和所述第二实体之间的关系之前，还包括：

训练所述知识图谱的所述嵌入式模型。