machine-learning - model.save() 保存的是最后一个 epoch 的模型还是最好的 epoch 的模型？

Question

这个单线用于保存keras深度学习神经网络模型。

model.save('my_model.h5')

model.save()保存的是最后一个epoch的模型还是最好的epoch的模型？有时，最后一个纪元并不能提高性能。

Answer 1

它将模型保存为其准确的当前状态。如果此语句是在 Model#fit 方法完成之后，则它代表最后一个纪元。

对于最佳时期(假设最佳 == 最小损失或更高准确度)，您可以使用 ModelCheckpoint为此:

epochs = 100
# other parameters...

model.fit(x, y,
          epochs=epochs,
          validation_data=valid,
          verbose=2,
          callbacks=[
              TerminateOnNaN(),
              TensorBoard('./logs'),
              ModelCheckpoint('best.h5',
                              save_best_only=True),
              ...
          ])
# the model is holding the weights optimized for 100 epochs.
model.load_weights('best.h5')  # load weights that generated the min val loss.