cogdl.tasks.node_classification

Module Contents

Classes

NodeClassification

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

Bases: cogdl.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', logits=None)[source]