Pytorch和torch_geometric一些代码的细节理解
1、GCN等图神经网络的输入参数对应关系
model部分内容
forward部分内容
输入对应的分别是每个节点的自身特征,图中边的关系,以及边权的值。
需要注意的是,边权值这一项常常被忽略,但实际运行中确有很大意义。
本来五个模型的ROC曲线基本一致,在origin_model上取消边权值一项后训练效果迅速下降。
2、关于torch_tensor,numpy.array的使用方法
torc...
原创
2020-04-29 11:13:25 ·
1503 阅读 ·
1 评论