如何使用Keras将检查点保存到路径?

5q4ezhmt  于 2023-01-30  发布在  其他
关注(0)|答案(1)|浏览(122)

如何在Keras中保存检查点并将其用于回调?
该文件需要为.h5模型。

62lalag4

62lalag41#

您需要设置.h5模型的路径目录,并定义检查点的参数。
我的建议是monitor= val_acc并将保存_best_only设置为True。

filepath="model.h5"
checkpoint1 = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=True, mode='max')
save_best_only=True, mode='min')
callbacks_list = [checkpoint1]

相关问题