StopperTrainingCallback
- class StopperTrainingCallback(stopper: Stopper, *, triples_factory: CoreTriplesFactory, last_best_epoch: int | None = None, best_epoch_model_file_path: Path | None)[source]
Bases:
TrainingCallback
An adapter for the
pykeen.stopper.Stopper
.Initialize the callback.
- Parameters:
stopper (Stopper) – the stopper
triples_factory (CoreTriplesFactory) – the triples factory used for saving the state
last_best_epoch (int | None) – the last best epoch
best_epoch_model_file_path (pathlib.Path | None) – the path under which to store the best model checkpoint
Methods Summary
post_epoch
(epoch, epoch_loss, **kwargs)Call after epoch.
Methods Documentation