动手学PyG(二)PyG中包含的常用Benchmark数据集

本文介绍了PyG库中包含的多种常用图数据集,如Planetoid系列和图分类数据集,并以ENZYMES数据集为例,详细说明了如何下载和访问数据,以及如何利用列表切片进行数据划分。同时提到了Cora数据集在无监督节点分类任务中的应用及其特殊属性。
摘要由CSDN通过智能技术生成

PyG中常用的Benchmark数据集


本文主要参考了 PyG英文文档

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
<
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值