我有一个现有的模型,并希望为其添加其他指标。 Keras metrics page表示指标是在编译时添加的,但我想在加载后添加它们(部分因为model.load_model()
似乎只加载了第一个指标,而且因为我有新指标我想要首先尝试现有的模型)。这可能吗?
答案 0 :(得分:4)
您可以根据需要多次编译模型。这不会改变它的重量。
因此,您可以加载模型,然后使用指标进行编译。
重新编译模型的唯一可能问题是重置优化器状态。
答案 1 :(得分:3)
只需添加此要点即可为您提供方便,快捷的复制/粘贴答案:
from keras.models import load_model
model_path = 'path/to/your/old_model.h5'
new_metrics = [<metrics to add>...] # for example ['binary_accuracy']
model = load_model(model_path)
model.compile(optimizer=model.optimizer,
loss=model.loss,
metrics=model.metrics+new_metrics)
如果未使用上述情况,请确保将所需的参数添加到编译函数。