cogdl.trainers.supervised_trainer

Module Contents

Classes

SupervisedTrainer	Helper class that provides a standard way to create an ABC using
SupervisedHeterogeneousNodeClassificationTrainer	Helper class that provides a standard way to create an ABC using
SupervisedHomogeneousNodeClassificationTrainer	Helper class that provides a standard way to create an ABC using

class cogdl.trainers.supervised_trainer.SupervisedTrainer

Bases: cogdl.trainers.base_trainer.BaseTrainer, abc.ABC

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

abstract fit(self) → None

abstract predict(self) → Any

class cogdl.trainers.supervised_trainer.SupervisedHeterogeneousNodeClassificationTrainer

Bases: cogdl.trainers.base_trainer.BaseTrainer, abc.ABC

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

abstract fit(self, model: cogdl.models.supervised_model.SupervisedHeterogeneousNodeClassificationModel, dataset: cogdl.data.Dataset) → None

class cogdl.trainers.supervised_trainer.SupervisedHomogeneousNodeClassificationTrainer

Bases: cogdl.trainers.base_trainer.BaseTrainer, abc.ABC

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

abstract fit(self, model: cogdl.models.supervised_model.SupervisedHomogeneousNodeClassificationModel, dataset: cogdl.data.Dataset) → None