tasks.heterogeneous_node_classification

Module Contents

Classes

HeterogeneousNodeClassification

Heterogeneous Node classification task.

class tasks.heterogeneous_node_classification.HeterogeneousNodeClassification(args)[source]

Bases: tasks.BaseTask

Heterogeneous Node classification task.

static add_args(parser)[source]

Add task-specific arguments to the parser.

train(self)[source]
_train_step(self)[source]
_test_step(self, split='val')[source]