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