停止训练回调

class StopperTrainingCallback(stopper: Stopper, *, triples_factory: CoreTriplesFactory, last_best_epoch: int | None = None, best_epoch_model_file_path: Path | None)[source]

基础类: TrainingCallback

一个用于pykeen.stopper.Stopper的适配器。

初始化回调。

Parameters:

  • stopper (Stopper) – 停止器

  • triples_factory (CoreTriplesFactory) – 用于保存状态的三元组工厂

  • last_best_epoch (int | None) – 最后一个最佳周期

  • best_epoch_model_file_path (pathlib.Path | None) – 存储最佳模型检查点的路径

方法总结

post_epoch(epoch, epoch_loss, **kwargs)

在epoch之后调用。

方法文档

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

在epoch之后调用。

Parameters:
Return type:

无