cogdl.tasks.node_classification
¶
Module Contents¶
Classes¶
Node classification task. |
-
class
cogdl.tasks.node_classification.
NodeClassification
(args, dataset=None, model: Optional[SupervisedHomogeneousNodeClassificationModel] = None)[source]¶ Bases:
cogdl.tasks.BaseTask
Node classification task.