cogdl.trainers.sampled_trainer

Module Contents

Classes

SampledTrainer

SAINTTrainer

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]
class cogdl.trainers.sampled_trainer.SAINTTrainer(args)[source]

Bases: cogdl.trainers.sampled_trainer.SampledTrainer

static build_trainer_from_args(args)[source]
sampler_from_args(self, args)[source]
fit(self, model: cogdl.models.supervised_model.SupervisedHeterogeneousNodeClassificationModel, dataset: cogdl.data.Dataset)[source]
_train_step(self)[source]
_test_step(self, split='val')[source]