cogdl.trainers.sampled_trainer
¶
Module Contents¶
Classes¶
-
class
cogdl.trainers.sampled_trainer.
SampledTrainer
[source]¶ Bases:
cogdl.trainers.supervised_trainer.SupervisedHeterogeneousNodeClassificationTrainer
-
abstract
fit
(self, model: cogdl.models.supervised_model.SupervisedHeterogeneousNodeClassificationModel, dataset: cogdl.data.Dataset)[source]¶
-
abstract
-
class
cogdl.trainers.sampled_trainer.
SAINTTrainer
(args)[source]¶ Bases:
cogdl.trainers.sampled_trainer.SampledTrainer
-
fit
(self, model: cogdl.models.supervised_model.SupervisedHeterogeneousNodeClassificationModel, dataset: cogdl.data.Dataset)[source]¶
-