一种基于对比学习的链接推理方法

本发明公开了一种基于对比学习的链接推理方法，属于数据处理技术领域。基于对比学习的链接推理方法，包括以下步骤：向已经训练好的图神经网络模型中输入已知的节点，得到节点的后验概率；对所有的节点按照有边和无边构建有边节点对和无边节点对；根据节点的后验概率对有边节点对和无边节点对分别进行特征处理；构建链接推理器，向链接推理器中输入有边节点对特征和/或无边节点对特征对链接推理器进行训练。采用本发明所述的基于对比学习的链接推理方法，通过链接推理器对未知节点是否有链接进行准确判断，可为推测关系等提供技术支持。