使用以下链接:
https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/ModelCheckpoint?version=stable
我无法确定monitor
参数的有效参数输入。
默认情况下,它设置为'val_loss'
。
在哪里可以找到该参数可接受的输入的列表?
对于我的问题有更多的背景信息,我试图设置一个检查点,以在n个历元之后保存“最佳”模型。但是“最佳”模型由monitor
参数确定-页面上的文档似乎未显示“可接受”输入。我想基于最平衡的精度/召回率(F1分数)保存“最佳”模型。
答案 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'
。