使用tf.keras.callbacks.ModelCheckpoint,“监视”参数的有效字符串值是什么?

时间:2019-12-19 23:18:26

标签: python-3.x tensorflow tensorflow2.0 tf.keras

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

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

对于我的问题有更多的背景信息,我试图设置一个检查点,以在n个历元之后保存“最佳”模型。但是“最佳”模型由monitor参数确定-页面上的文档似乎未显示“可接受”输入。我想基于最平衡的精度/召回率(F1分数)保存“最佳”模型。

1 个答案:

答案 0 :(得分:1)

monitor的{​​{1}} arg希望您提供一个字符串,该字符串必须是度量标准的名称或丢失的名称,例如,如果您的编译方法看起来像这样

ModelCheckpoint

model.compile(loss='mse', optimizer='sgd', metrics=['mae', 'accuracy']) arg的有效字符串为:

monitor'train_loss''val_loss''train_mae''val_mae''train_accuracy'