[发明专利]一种基于图神经网络进行预测的方法、系统和装置

摘要

本说明书提供一种基于图神经网络进行预测的方法、系统和装置，所述方法包括：将多个节点的特征向量输入转换模型，获得多个节点的转换向量；针对每一个邻居节点，将待预测节点的转换向量、邻居节点的转换向量，以及连接待预测节点和邻居节点的边的特征向量拼接，并将拼接得到的第一拼接向量输入传播模型，获取邻居节点的传播向量；基于至少一个邻居节点的传播向量以及待预测节点的转换向量，聚合获取待预测节点的表示向量；用待预测节点的表示向量替代待预测节点的转换向量，迭代传播和聚合，得到待预测节点的新的表示向量；将待预测节点的新的表示向量输入预测模型，获取待预测节点的预测结果。