将指标添加到Keras中的现有模型

时间:2017-05-30 16:07:49

标签: machine-learning keras metrics

我有一个现有的模型,并希望为其添加其他指标。 Keras metrics page表示指标是在编译时添加的,但我想在加载后添加它们(部分因为model.load_model()似乎只加载了第一个指标,而且因为我有新指标我想要首先尝试现有的模型)。这可能吗?

2 个答案:

答案 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)

如果未使用上述情况,请确保将所需的参数添加到编译函数。