BA2MotifDataset

class dgl.data.BA2MotifDataset(raw_dir=None, force_reload=False, verbose=True, transform=None)

基类：DGLBuiltinDataset

来自 图神经网络的参数化解释器 的 BA-2motifs 数据集

这是一个用于图分类的合成数据集。它通过按以下顺序执行以下步骤生成。

• 构建 1000 个基础 Barabási–Albert (BA) 图。

• 将房屋结构的子图附加到一半的基础 BA 图上。

• 将五节点环形子图附加到其余的基础 BA 图上。

• 根据附加的子图类型将每个图分配到两个类别之一。

参数：

• raw_dir (str, optional) – 原始文件目录，用于下载和存储数据。默认值：~/.dgl/

• force_reload (bool, optional) – 是否重新加载数据集。默认值：False

• verbose (bool, optional) – 是否打印进度信息。默认值：True

• transform (callable, optional) – 一个转换函数，接受一个 DGLGraph 对象并返回一个转换后的版本。在每次访问之前，该 DGLGraph 对象都会被转换。默认值：None

num_classes

图类别数量

类型：

int

示例

>>> from dgl.data import BA2MotifDataset
>>> dataset = BA2MotifDataset()
>>> dataset.num_classes
2
>>> # Get the first graph and its label
>>> g, label = dataset[0]
>>> feat = g.ndata['feat']

__getitem__(idx)

获取指定索引的数据对象。

__len__()

数据集中示例的数量。