加载keras模型h5未知度量

Question

我培训了keras CNN，监控以下指标：

METRICS = [
  TruePositives(name='tp'),
  FalsePositives(name='fp'),
  TrueNegatives(name='tn'),
  FalseNegatives(name='fn'), 
  BinaryAccuracy(name='accuracy'),
  Precision(name='precision'),
  Recall(name='recall'),
  AUC(name='auc'),
 ]

然后是model.compile：

 model.compile(optimizer='nadam', loss='binary_crossentropy',
         metrics=METRICS)

它工作得很好，我保存了我的h5型号（model.h5）

现在我已经下载了该模型，我想在其他脚本中使用它导入模型，包括：

 from keras.models import load_model
 model = load_model('model.h5')
 model.predict(....)

但在运行过程中，编译器返回：

 ValueError: Unknown metric function: {'class_name': 'TruePositives', 'config': {'name': 'tp', 'dtype': 'float32', 'thresholds': None}}

我应该如何处理这个问题

先谢谢你

Answer 1

看起来您正在使用tensorflow教程。我也使用了这些精确的指标，也遇到了同样的问题。对我来说，有效的方法是用compile = False加载模型，然后用自定义度量编译它。然后您应该能够按预期使用model.predict(....)

import keras

model = keras.models.load_model('model.h5', compile = False)

METRICS = [
      keras.metrics.TruePositives(name='tp'),
      keras.metrics.FalsePositives(name='fp'),
      keras.metrics.TrueNegatives(name='tn'),
      keras.metrics.FalseNegatives(name='fn'), 
      keras.metrics.BinaryAccuracy(name='accuracy'),
      keras.metrics.Precision(name='precision'),
      keras.metrics.Recall(name='recall'),
      keras.metrics.AUC(name='auc'),
]

model.compile(optimizer = keras.optimizers.Adam(learning_rate=1e-4),
              loss = 'binary_crossentropy',
              metrics = METRICS
             )