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

post_epoch(epoch: int, epoch_loss: float, **kwargs: Any) None[source]

Call after epoch.

Parameters:
Return type:

None