KarateClubDataset

class dgl.data.KarateClubDataset(transform=None)[source]

Bases: DGLDataset

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

扎卡里的空手道俱乐部是一个大学空手道俱乐部的社交网络,由韦恩·W·扎卡里在论文《小群体中的冲突与分裂的信息流模型》中描述。该网络在2002年被米歇尔·吉文和马克·纽曼使用后,成为网络中社区结构的热门示例。官方网站:http://konect.cc/networks/ucidata-zachary/

Karate Club 数据集统计:

  • 节点数:34

  • 边数:156

  • 班级数量:2

Parameters:

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.

num_classes

节点类的数量

Type:

int

示例

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

获取图形对象

Parameters:

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

Returns:

图结构和标签。

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

Return type:

dgl.DGLGraph

__len__()[source]

数据集中图的数量。