easygraph.datasets.hypergraph.coauthorship module

class easygraph.datasets.hypergraph.coauthorship.CoauthorshipCora(data_root: str | None = None)

Bases: BaseData

The Co-authorship Cora dataset is a citation network dataset for vertex classification task. More details see the HyperGCN paper.

The content of the Co-authorship Cora dataset includes the following:

• num_classes: The number of classes: \(7\).

• num_vertices: The number of vertices: \(2,708\).

• num_edges: The number of edges: \(1,072\).

• dim_features: The dimension of features: \(1,433\).

• features: The vertex feature matrix. torch.Tensor with size \((2,708 \times 1,433)\).

• edge_list: The edge list. List with length \(1,072\).

• labels: The label list. torch.LongTensor with size \((2,708, )\).

• train_mask: The train mask. torch.BoolTensor with size \((2,708, )\).

• val_mask: The validation mask. torch.BoolTensor with size \((2,708, )\).

• test_mask: The test mask. torch.BoolTensor with size \((2,708, )\).

Parameters:

data_root (str, optional) – The data_root has stored the data. If set to None, this function will auto-download from server and save into the default direction ~/.dhg/datasets/. Defaults to None.

Attributes:

content

Return the content of the dataset.

Methods

fetch_files(files)	Download and check the files if they are not exist.
needs_to_load(item_name)	Return whether the item_name of the dataset needs to be loaded.
raw(key)	Return the key of the dataset with un-preprocessed format.

class easygraph.datasets.hypergraph.coauthorship.CoauthorshipDBLP(data_root: str | None = None)

Bases: BaseData

The Co-authorship DBLP dataset is a citation network dataset for vertex classification task. More details see the HyperGCN paper.

The content of the Co-authorship DBLP dataset includes the following:

• num_classes: The number of classes: \(6\).

• num_vertices: The number of vertices: \(41,302\).

• num_edges: The number of edges: \(22,363\).

• dim_features: The dimension of features: \(1,425\).

• features: The vertex feature matrix. torch.Tensor with size \((41,302 \times 1,425)\).

• edge_list: The edge list. List with length \(22,363\).

• labels: The label list. torch.LongTensor with size \((41,302, )\).

• train_mask: The train mask. torch.BoolTensor with size \((41,302, )\).

• val_mask: The validation mask. torch.BoolTensor with size \((41,302, )\).

• test_mask: The test mask. torch.BoolTensor with size \((41,302, )\).

Parameters:

data_root (str, optional) – The data_root has stored the data. If set to None, this function will auto-download from server and save into the default direction ~/.dhg/datasets/. Defaults to None.

Attributes:

content

Return the content of the dataset.

Methods

fetch_files(files)	Download and check the files if they are not exist.
needs_to_load(item_name)	Return whether the item_name of the dataset needs to be loaded.
raw(key)	Return the key of the dataset with un-preprocessed format.