Keras: Fehler beim Laden Modell mit benutzerdefinierten Verlustfunktion

Question

ich eine benutzerdefinierte Verlustfunktion wie folgt definieren:

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) 

Wenn ich das Modell in einer hdf5 Datei speichern und dann versuchen, es mit load_model zu laden, habe ich folgende Fehler Informationen:

ValueError: Unknown loss function:mse_weighted

Wer weiß, wie dieses Problem zu beheben?

Danke!

Answer 1

Dies wird hier gelöst ein Benutzerwörterbuch Objekt load_model indem:

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