cogdl.models.supervised_model

Module Contents

Classes

SupervisedModel	Helper class that provides a standard way to create an ABC using
SupervisedHeterogeneousNodeClassificationModel	Helper class that provides a standard way to create an ABC using
SupervisedHomogeneousNodeClassificationModel	Helper class that provides a standard way to create an ABC using

class cogdl.models.supervised_model.SupervisedModel

Bases: cogdl.models.BaseModel, abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

abstract loss(self, data: Any) → Any

class cogdl.models.supervised_model.SupervisedHeterogeneousNodeClassificationModel

Bases: cogdl.models.BaseModel, abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

abstract loss(self, data: Any) → Any

evaluate(self, data: Any, nodes: Any, targets: Any) → Any

static get_trainer(taskType: Any, args: Any) → Optional[Type[SupervisedHeterogeneousNodeClassificationTrainer]]

class cogdl.models.supervised_model.SupervisedHomogeneousNodeClassificationModel

Bases: cogdl.models.BaseModel, abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

abstract loss(self, data: Any) → Any

abstract predict(self, data: Any) → Any

static get_trainer(taskType: Any, args: Any) → Optional[Type[SupervisedHomogeneousNodeClassificationTrainer]]