Keras：带有自定义丢失功能的错误加载模型

Question

我定义了一个自定义丢失函数，如下所示：

weight_for_hierarchical_error = K.variable(np.ones(16))

def mse_weighted(y_true, y_pred):
    return K.mean(weight_for_hierarchical_error * K.square(y_pred - y_true), axis=-1)

当我将模型保存在hdf5文件中然后尝试使用load_model加载它时，我收到以下错误信息：

ValueError: Unknown loss function:mse_weighted

有谁知道如何解决这个问题？

谢谢！

Answer 1

这可以通过将自定义词典对象传递给load_model来解决：

https://github.com/fchollet/keras/issues/5916 https://github.com/fchollet/keras/issues/3977