我有一个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)
。有趣的是,当我通过命令外壳再次运行同一命令时,不会出现此错误,它仅在第一次运行时发生。这是错误吗?