图神经网络预训练 (4) - 节点属性预测 Attribute Prediction + 监督学习 代码
我们继续剖析StrategiesforPre-trainingGraphNeuralNetworks一文。上一文中介绍了子结构预测的预训练方法(ContextPrediction)。对于一个多层的深度学习模型,在分子图上训练主模型,在子图上训练层数较少的子模型,限制(损失是)主模型上与子模型的嵌入向量相似,保证子结构环境相似的节点在随着模型的层数增加时仍能保持相似的嵌入向量,意味着化学环境相似的结