torch_geometric.datasets.KarateClub
- class KarateClub(transform: Optional[Callable] = None)[source]
Bases:
InMemoryDataset
Zachary’s karate club network from the “An Information Flow Model for Conflict and Fission in Small Groups” paper, containing 34 nodes, connected by 156 (undirected and unweighted) edges. Every node is labeled by one of four classes obtained via modularity-based clustering, following the “Semi-supervised Classification with Graph Convolutional Networks” paper. Training is based on a single labeled example per class, i.e. a total number of 4 labeled nodes.
- 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
)
STATS:
#nodes
#edges
#features
#classes
34
156
34
4