使用自定义指标加载keras模型

时间:2019-08-07 18:08:02

标签: python

我用一个名为“ correlation_coefficient”的自定义指标来保存一个keras模型。 当我尝试按以下方式加载保存的模型时:

model = load_model('model.h5', custom_objects={'correlation_coefficient':correlation_coefficient})

出现以下错误:

FailedPreconditionError: 2 root error(s) found.
  (0) Failed precondition: Attempting to use uninitialized value metrics_7/correlation_coefficient/pearson_r/variance_labels/mean_label
     [[{{node metrics_7/correlation_coefficient/pearson_r/variance_labels/mean_label/read}}]]
     [[loss_7/mul/_16917]]
  (1) Failed precondition: Attempting to use uninitialized value metrics_7/correlation_coefficient/pearson_r/variance_labels/mean_label
     [[{{node metrics_7/correlation_coefficient/pearson_r/variance_labels/mean_label/read}}]]
0 successful operations.
0 derived errors ignored.

注意:我使用Kaggle来运行代码

0 个答案:

没有答案