lenskit.graphs.lightgcn.PairwiseLightGCNTrainer#
- class lenskit.graphs.lightgcn.PairwiseLightGCNTrainer(scorer, data, options)#
Bases:
LightGCNTrainerProtocol implemented by iterative trainers for models. Models that implement
UsesTrainerwill return an object implementing this protocol from theircreate_trainer()method.This protocol only defines the core aspects of training a model. Trainers should also implement
ParameterContainerto allow training to be checkpointed and resumed.It is also a good idea for the trainer to be pickleable, but the parameter container interface is the primary mechanism for checkpointing.
- Stability:
- Full (see Stability Levels).
- Parameters:
scorer (LightGCNScorer)
data (lenskit.data.Dataset)
options (lenskit.training.TrainingOptions)
- batch_loss(mb_edges, scores)#
- Parameters:
mb_edges (torch.Tensor)
scores (torch.Tensor)