keras - 如何在 Keras 中使用带有加权验证损失的 ModelCheckpoint()

Question

我在 Keras 训练一个 DNN，它具有高度不平衡的类。所以我在 fit_generator 中使用了 class_weight 来纠正这个问题。现在我想使用 ModelCheckpoint() 函数保存具有最低加权验证损失的模型。我正在尝试，但我无法弄清楚实现这一目标的方法。有人有一个简单的例子吗？

标签： keras

Answer 1

ModelCheckpoint("checkpoint.hdf5", monitor='val_loss', mode = 'min', verbose=1, save_best_only = True)
model.fit_genetor(....)

我想你是在问这段代码。