cogdl.tasks.multiplex_node_classification

Module Contents

Classes

MultiplexNodeClassification

Node classification task.

class cogdl.tasks.multiplex_node_classification.MultiplexNodeClassification(args, dataset=None, model=None)[source]

Bases: cogdl.tasks.BaseTask

Node classification task.

static add_args(parser)[source]

Add task-specific arguments to the parser.

train(self)[source]