keras.callbacks.ModelCheckpoint()

每个纪元后保存模型。
filepath可以包含命名格式化选项,它将填充日志中的纪元和键的值(在on_epoch_end中传递)。 例如:如果filepath是权重。{epoch:02d} - {val_loss:.2f} .hdf5,那么模型检查点将与文件名中的纪元号和验证丢失一起保存。

def __init__(self,
             filepath: Any,
             monitor: str = 'val_loss',
             verbose: int = 0,
             save_best_only: bool = False,
             save_weights_only: bool = False,
             mode: str = 'auto',
             period: int = 1) -> None

参数

  • filepath:string,保存模型文件的路径。
  • monitor:监控:要监控的数量。
  • verbose详细:详细模式,0或1。
  • save_best_only:如果save_best_only = True,则不会覆盖根据监控数量的最新最佳模型。
  • save_weights_only:如果为True,则只有模型的权重
    保存(model.save_weights(filepath)),否则保存完整模型(model.save(filepath))。
  • mode:{auto,min,max}之一。 如果save_best_only =
    True,则根据监控数量的最大化或最小化来决定覆盖当前保存文件。
    对于val_acc,这应该是max,对于val_loss,这应该是min等。在自动模式下,从监控量的名称自动推断方向。
  • period:检查点之间的间隔(时期数)。