通过torch_geometric.data.DataLoader可以方便地使用 mini-batch。 fromtorch_scatterimportscatter_meanfromtorch_geometric.datasetsimportTUDatasetfromtorch_geometric.loaderimportDataLoaderdataset=TUDataset(root='/tmp/ENZYMES',name='ENZYMES',use_node_attr=True)loader=DataLoader(dataset,batch_size=32,shuffle=True)fo...
官方仓库:https://github.com/pyg-team/pytorch_geometric 官方文档:PyG Documentationtorch_geometric.data .Data(同构图)的用法labelnamedescribeshape√data.keys返回属性名列表√data.y真实标签√data.x节点…
首先,导入torch_geometric中的数据集模块,并选择一个数据集进行演示: ```python import torch import torch_geometric from torch_geometric.datasets import TUDataset #选择一个数据集作为示例 dataset = TUDataset(root='/path_to_dataset', name='ENZYMES') ``` 在以上代码中,我们加载了一个图分类任务的数据集...
I was having the same issue with torch version 1.11.0 when importing torch_geometric datasets from torch_geometric.datasets import TUDataset I was able to resolve with by uninstalling torch-geometric and all its dependencies with: pip uninstall torch-geometric torch-scatter torch-sparse torch-cluster...
from torch_geometric.datasets import TUDataset # 导入数据集 dataset = TUDataset( # 指定数据集的存储位置 # 如果指定位置没有相应的数据集 # PyG会自动下载 root='../data/ENZYMES', # 要使用的数据集 name='ENZYMES', ) # 数据集的长度 print(len(dataset)) # 数据集的类别数 print(dataset.num_clas...