PyG中包含了大量常用的benchmark数据集,如所有Planetoid数据集(Cora, Citeseer, Pubmed)。来自http://graphkernels.cs.tu-dortmund.de/的所有图分类数据集,和其简洁版,QM7和QM9数据集等等。
数据集的下载和使用非常简单,下面以ENZYMES数据集(包含600个图,分为6类)为例做介绍。
from torch_geometric.datasets import TUDataset
dataset = TUDataset(root='/tmp/ENZYMES', name='ENZYMES')
>>> ENZYMES(600)
len(dataset)
>>> 600
dataset.num_classes
>>> 6
dataset.num_node_features
>>> 3
<