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