在Keras中使用EarlyStopping回调时如何获得最佳模型？

Question

我正在使用EarlyStopping基于val_acc和的Keras训练神经网络patience=0。下降后EarlyStopping立即停止训练val_acc。

但是，我获得的最终模型不是最佳模型，即具有最高模型的模型val_acc。但是我宁愿有一个对应于之后的时代的模型，也就是那个对应于val_acc最好的模型的模型，它比最好的模型低了一点，并导致了早期停止！

我如何获得最好的？

我试图通过回叫使用保存最佳模型：

ModelCheckpoint(filepath='best_model.h5', monitor='val_loss', save_best_only=True)]

但我得到相同的结果。

Answer 1

如果要保存最高的准确性，则应设置检查点monitor='val_acc'，该检查点将自动保存在最高的准确性上。最低的损失不一定与最高的精度相对应。您还可以设置verbose=1查看要保存的模型以及原因。