Task¶
The task means downstream task. So it is related to dataset closely.
Included Object:¶
Dataset, containing the heterogeneous graph as DGLGraph, as well as node/edge features and additional dataset-specific information.
Evaluation Metric, a callable object which takes in the prediction and the ground truth label, and returns an output.
Currently we will implement evaluation metrics for node classification and link prediction.
Training Loss, a callable object which takes in the prediction and the ground truth as well as any other necessary information and returns an output. It must be differentiable.
Ranking loss for self-supervision, or contrastive learning losses, all fall into this category.
Training-validation-test Split.
Method:¶
..func:
get_graph()
return the DGLGraph
get_loss_fn()
train with a full_batch graph
evaluate()
Evaluate Metric
get_labels()
return labels
Supported task¶
Node classification
Link prediction