mlx_graphs.datasets.KarateClubDataset

class mlx_graphs.datasets.KarateClubDataset

Zachary’s Karate Club netowork dataset from An Information Flow Model for Conflict and Fission in Small Groups. This is a simple dataset for node classification. The graph has 34 nodes and 156 (undirected) edges. Each node belongs to one of 2 classes.

__init__()

Methods

__init__()
download()	Download the dataset at self.raw_path.
load()	Load the processed dataset
process()	Process the dataset and store data in self.data
save()	Save the processed dataset

Attributes

name	Name of the dataset
num_edge_classes	Returns the number of edge classes to predict.
num_edge_features	Returns the number of edge features.
num_graph_classes	Returns the number of graph classes to predict.
num_graph_features	Returns the number of graph features.
num_graphs	Returns the number of graphs in the dataset.
num_items	Returns the number of items in the dataset.
num_node_classes	Returns the number of node classes to predict.
num_node_features	Returns the number of node features.
processed_path	The path where raw files are stored.
raw_path	The path where raw files are stored.