将加载的Keras分类器与自定义指标功能一起使用时出错

我有一个keras模型,该模型使用自定义函数进行度量:

model.compile(optimizer = tf.keras.optimizers.Adam(),loss = 'binary_crossentropy',metrics = ['accuracy',roc_auc])

该函数运行正常,模型行为符合预期。但是,当通过model.save()保存模型,然后通过model.load_model()加载模型时,运行以下代码:ValueError: Unknown metric function:roc_auc时得到model.predict(X)。有趣的是,当我通过命令外壳再次运行同一命令时,不会出现此错误,它仅在第一次运行时发生。这是错误吗?

john_anson 回答:将加载的Keras分类器与自定义指标功能一起使用时出错

您是说keras.models.load_model(path)吗?拥有model.load_model()听起来很奇怪。

您可能缺少custom_objects = {'roc_auc': roc_auc}中的参数load_model。如果Keras不知道roc_auc的含义,则无法创建模型。

本文链接:https://www.f2er.com/3124374.html

大家都在问