openhgnn.tasks.EdgeClassification

class EdgeClassification(args)

Edge classification tasks.

dataset

Task-related dataset

Type:

EdgeClassificationDataset

evaluator

offer evaluation metric

Type:

Evaluator

get_graph :

return a graph

get_loss_fn :

return a loss function