torch_geometric.datasets.KarateClub
- class KarateClub(transform: Optional[Callable] = None)[source]
Bases:
InMemoryDatasetZachary的空手道俱乐部网络来自“小群体中的冲突与分裂的信息流模型”论文,包含34个节点,通过156条(无向且无权重)边连接。每个节点通过基于模块化的聚类方法标记为四个类别之一,遵循“图卷积网络的半监督分类”论文。训练基于每个类别的一个标记示例,即总共4个标记节点。
- Parameters:
transform (callable, optional) – A function/transform that takes in an
torch_geometric.data.Dataobject and returns a transformed version. The data object will be transformed before every access. (default:None)
统计:
#节点
#edges
#特性
#classes
34
156
34
4