openhgnn.dataset.NodeClassificationDataset¶

class NodeClassificationDataset(*args, **kwargs)[source]¶

The class NodeClassificationDataset is a base class for datasets which can be used in task node classification. So its subclass should contain attributes such as graph, category, num_classes and so on. Besides, it should implement the functions get_labels() and get_split().

g¶

The heterogeneous graph.

Type:: dgl.DGLHeteroGraph

category¶

The category(or target) node type need to be predict. In general, we predict only one node type.

Type:: str

num_classes¶

The target node will be classified into num_classes categories.

Type:: int

has_feature¶

Whether the dataset has feature. Default False.

Type:: bool

multi_label¶

Whether the node has multi label. Default False. For now, only HGBn-IMDB has multi-label.

Type:: bool