定义自定义损失函数后出现keras "unknown loss function"错误

Question

我在loss.py文件中定义了keras的新损失函数。我关闭并重新启动 anaconda 提示符，但出现了 ValueError: ('Unknown loss function', ':binary_crossentropy_2')。我在 Windows 10 上使用 python2.7 和 anaconda 运行 keras。

我通过在编译模型的Python文件中添加损失函数来临时解决这个问题。

Answer 1

在 Keras 中，我们必须在 load_model 函数中传递自定义函数:

def my_custom_func():
    # your code
    return
from keras.models import load_model
model = load_model('my_model.h5', custom_objects={'my_custom_func':                   
my_custom_func})