2017-12-07 6 views
0

Ich möchte ein Keras-Modell, das eine Beschreibung enthält (kann eine einfache Zeichenfolge sein) schreiben, so dass man leichter verfolgen kann, was eine Datei enthält.Saving Modell mit Beschreibung in Keras

Gibt es eine einfache Möglichkeit, dies zu tun? Ich kann mit

model.save(filename) 
[...] 
model = keras.load_model(filename) 

ein Modell speichern und laden, was ich möchte, zu tun ist, etwas zu den Auswirkungen von

model.save(filename, description=string) 
[...] 
model, description = keras.load_model(filename) 

Irgendwelche Ideen auf, was der einfachste Weg, dies zu erreichen wäre?

Antwort

0

Momentan gibt es keine Möglichkeit, dies mit der aktuellen Version model.save im Keras zu tun. Es speichert nur die Konfiguration des Modells() wie loss function und metrics und load_model gibt nur zurück, was gespeichert wurde.

Eine unangenehme Lösung für dieses Problem ist das Hinzufügen eines manuell description Variable zu der Klasse models.py in Keras und folglich, fügt die erforderlichen Codezeilen zu den save_model und load_model Funktionen. Auch dies ist eine schreckliche Lösung!