tasks.node_classification

Module Contents

Classes

NodeClassification

Node classification task.
class tasks.node_classification.NodeClassification(args, dataset=None, model=None)[source]

Bases: tasks.BaseTask

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]