Dataset

openhgnn.dataset.BaseDataset

openhgnn.dataset.NodeClassificationDataset

The class NodeClassificationDataset is a base class for datasets which can be used in task node classification.

openhgnn.dataset.LinkPredictionDataset

metric: Accuracy, multi-label f1 or multi-class f1.

openhgnn.dataset.RecommendationDataset

openhgnn.dataset.AcademicDataset

openhgnn.dataset.HGBDataset

openhgnn.dataset.OHGBDataset

openhgnn.dataset.GTNDataset

GTN Dataset.

openhgnn.dataset.AsLinkPredictionDataset

Repurpose a dataset for link prediction task.

openhgnn.dataset.AsNodeClassificationDataset

Repurpose a dataset for a standard semi-supervised transductive node prediction task.

openhgnn.dataset.EdgeClassificationDataset

The class EdgeClassificationDataset is a base class for datasets which can be used in task edge classification.