一种基于节点嵌入和元路径嵌入的链路预测方法和装置

本发明公开了一种基于节点嵌入和元路径嵌入的链路预测方法和装置，属于链路预测领域，所述方法为：获取图数据；其中，所述图数据包括第一节点特征矩阵和第一边索引；基于所述第一节点特征矩阵和第一边索引，通过多头注意力机制聚合邻居节点特征，得到节点嵌入矩阵；基于所述节点嵌入矩阵，通过特征相似性和个性化PageRank动态生成元路径，并通过多头注意力机制聚合元路径内部的节点特征，得到元路径嵌入矩阵；基于所述节点嵌入矩阵和元路径嵌入矩阵，通过多层感知器将节点特征和元路径特征映射到一维输出，得到所述图数据的链路预测结果。因此，通过实施本发明，能够融合图数据的局部信息和全局信息，提高链路预测的准确性。