openhgnn.tasks.LinkPrediction

class LinkPrediction(args)

Link prediction tasks.

dataset

Task-related dataset

Type:

NodeClassificationDataset

evaluator

offer evaluation metric

Type:

Evaluator

get_graph :

return a graph

get_loss_fn :

return a loss function