图神经网络训练时把整个图的信息都放进去训练,然后只计算训练集节点的损失吗
时间: 2023-12-19 20:06:44 浏览: 187
zzzsjwl.rar_self-organizing map_神经网络 图像_神经网络图_自组织网络_训练
在图神经网络的训练中,一般会使用节点分类、链接预测等任务作为目标任务。对于节点分类任务,我们会将整个图的信息作为输入,然后仅计算训练集中节点的损失。这是因为我们的目标是对训练集中的节点进行分类,而不是对整个图进行分类。对于链接预测任务,也是类似的,我们会将整个图的信息作为输入,然后仅计算训练集中存在的边的损失。这样可以有效地利用图的结构信息,提高模型的泛化能力。
阅读全文