BA2Motif数据集

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

Bases: DGLBuiltinDataset

BA-2motifs 数据集来自 Parameterized Explainer for Graph Neural Network

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

构建1000个基础的Barabási–Albert (BA)图。
将房屋结构的网络基元附加到一半的基础BA图上。
将五节点循环图案附加到其余的BA基础图上。
根据附加的模体类型，将每个图分配到两个类别之一。

Parameters:

raw_dir (str, optional) – 用于下载和存储数据的原始文件目录。默认值：~/.dgl/
force_reload (bool, optional) – Whether to reload the dataset. Default: False
verbose (bool, optional) – Whether to print progress information. Default: True
transform (callable, optional) – A transform that takes in a DGLGraph object and returns a transformed version. The DGLGraph object will be transformed before every access. Default: None

num_classes

图类数量

Type:: 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)[source]: 获取索引处的数据对象。

__len__()[source]: 数据集中的示例数量。