openhgnn.dataset.EdgeClassificationDataset¶
- class EdgeClassificationDataset(*args, **kwargs)[source]¶
The class EdgeClassificationDataset is a base class for datasets which can be used in task edge 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: