KarateClubDataset

class dgl.data.KarateClubDataset(transform=None)

基类: DGLDataset

用于节点分类的 Karate Club 数据集

Zachary 的空手道俱乐部是大学空手道俱乐部的社交网络，由 Wayne W. Zachary 在论文《An Information Flow Model for Conflict and Fission in Small Groups》中描述。2002 年，Michelle Girvan 和 Mark Newman 使用该网络后，它成为了网络社区结构的流行示例。官方网站：http://konect.cc/networks/ucidata-zachary/

Karate Club 数据集统计信息

• 节点数: 34

• 边数: 156

• 类别数: 2

参数：

transform (callable, optional) – 一个转换函数，接受一个 DGLGraph 对象并返回一个转换后的版本。该 DGLGraph 对象会在每次访问前进行转换。

num_classes

节点类别数

类型：

int

示例

>>> dataset = KarateClubDataset()
>>> num_classes = dataset.num_classes
>>> g = dataset[0]
>>> labels = g.ndata['label']

__getitem__(idx)

获取图对象

参数：

idx (int) – 项索引，KarateClubDataset 只有一个图对象

返回：

图结构和标签。

• ndata['label']: 真实标签

返回类型：

dgl.DGLGraph

__len__()

数据集中的图数量。