使用 tf.keras.callbacks.ModelCheckpoint，'monitor' 参数的有效字符串值是什么？

Question

使用以下链接： https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/ModelCheckpoint?version=stable 我无法确定 monitor 参数的有效参数输入。 默认设置为'val_loss'。

在哪里可以找到此参数的可接受输入列表？

有了更多关于我的问题的背景信息，我正在尝试设置一个检查点，以在 n 个 epoch 之后保存“最佳”模型。但是“最佳”模型由 monitor 参数确定 - 页面上的文档似乎没有显示“可接受的”输入。我想根据最平衡的精度/召回率（F1 分数）保存“最佳”模型。

Answer 1

ModelCheckpoint 的 monitor arg 期望您提供一个字符串，这需要是度量或损失的名称，例如，如果您的编译方法看起来像这样

model.compile(loss='mse', optimizer='sgd', metrics=['mae', 'accuracy'])

monitor arg 的有效字符串为：

'train_loss'、'val_loss'、'train_mae'、'val_mae'、'train_accuracy'、'val_accuracy'。