torch_geometric.datasets.KarateClub

class KarateClub(transform: Optional[Callable] = None)[source]

Bases: InMemoryDataset

Zachary的空手道俱乐部网络来自“小群体中的冲突与分裂的信息流模型”论文,包含34个节点,通过156条(无向且无权重)边连接。每个节点通过基于模块化的聚类方法标记为四个类别之一,遵循“图卷积网络的半监督分类”论文。训练基于每个类别的一个标记示例,总共4个标记节点。

Parameters:

transform (callable, optional) – A function/transform that takes in an torch_geometric.data.Data object and returns a transformed version. The data object will be transformed before every access. (default: None)

统计:

#节点

#edges

#特性

#classes

34

156

34

4