图神经网络系列-图应用-节点分类与关系推理-图分类任务
系列文章 图神经网络系列-图应用-节点分类与关系推理-Cora network论文引用案例代码实战 https://blog.csdn.net/duan_zhihua/article/details/115496679?spm=1001.2014.3001.5501
图神经网络系列KarateClub案例代码实战 https://blog.csdn.net/duan_zhihua/article/details/115468187?spm=1001.2014.3001.5501
目录
TU dataset 案例
- TU dataset 案例
- 数据并行处理
- GNN模型训练
基于多特蒙德大学收集的图形分类数据集TU dataset预测图分类。图分类最常见的任务是分子性质预测,分子表示为图,推断一个分子是否可以抑制病毒复制。
import torch
from torch_geometric