Keras ModelCheckpoint 监控多个值

Question

我想使用 Keras ModelCheckpoint 回调来监控几个参数（我有一个多任务网络）。只有一个回调有可能吗？还是我需要在很多回调中这样做？？

创建 ckechpoint ：

checkpointer = ModelCheckpoint(filepath='checkpoints/weights-{epoch:02d}.hdf5', monitor='val_O1_categorical_accuracy' , verbose=1, save_best_only=True, mode='max')

我要监控的第二个参数：val_O2_categorical_accuracy

在列表中这样做是行不通的。即

checkpointer = ModelCheckpoint(filepath='checkpoints/weights-{epoch:02d}.hdf5', monitor=['val_O1_categorical_accuracy','val_O2_categorical_accuracy'] , verbose=1, save_best_only=True, mode='max')

TypeError: unhashable type: 'list'

Answer 1

恐怕您将不得不在不同的情况下执行此操作。想想这里发生了什么 -

checkpointer = ModelCheckpoint(filepath='checkpoints/weights-{epoch:02d}.hdf5', monitor='val_O1_categorical_accuracy' , verbose=1, save_best_only=True, mode='max')

当您通过监视val_O1_categorical_accuracy 保存模型时，它会在伪代码中执行以下操作 -

for each epoch:
    check the val_O1_categorical_accuracy after updating weights
    if this metric is better in this epoch than the previous ones:
        save the model
    else
        pass

所以真正指定多个monitor 超出了范围。在这种情况下，它必须是一个非此即彼的选择，因为基于 monitor 指标，在其他相互冲突的模型中，只有一个模型可能是最好的。