我有一个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)。有趣的是，当我通过命令外壳再次运行同一命令时，不会出现此错误，它仅在第一次运行时发生。这是一个错误吗？
你的意思是keras.models.load_model(path)？听起来很奇怪model.load_model()。
你可能缺少的说法custom_objects = {'roc_auc': roc_auc}在load_model。如果Keras不知道什么roc_auc意思，则无法创建模型。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句