Anfügen von Klassenetiketten an ein Keras-Modell

Question

Ich verwende das Keras Sequential-Modell, um mehrere Klassenklassifizierer zu trainieren.

Bei der Auswertung gibt Keras einen Vektor der Vertraulichkeiten aus und ich kann die korrekte Klassen-ID aus der mit Argmax ableiten. Ich kann dann eine Nachschlagetabelle verwenden, um die tatsächliche Klassenbezeichnung (z. B. eine Zeichenfolge) zu erhalten.

Bis jetzt ist die Lösung, das trainierte Modell zu laden und dann eine Lookup-Tabelle separat zu laden. Da ich eine ganze Reihe von Klassifikatoren habe, würde ich es vorziehen, beide Strukturen in einer Datei zu behalten.

Also was ich suche ist eine Möglichkeit, den tatsächlichen Label-Lookup-Vektor in das Keras-Modell zu integrieren. Das würde mir erlauben, eine einzige Klassifizierungsdatei zu haben, die in der Lage ist, einige Eingabedaten zu nehmen und die korrekte Klassenbezeichnung für diese Daten zurückzugeben.

Eine Möglichkeit, dies zu lösen, wäre, sowohl das Modell als auch die Nachschlagetabelle in einem Tupel zu speichern und dieses Tupel in eine Beize zu schreiben, aber das scheint nicht sehr elegant zu sein.

Answer 1

Also habe ich mich selbst an einer Lösung versucht und das scheint zu funktionieren. Ich habe auf etwas einfacheres gehofft.

Das Öffnen der Modelldatei ein zweites Mal ist nicht wirklich optimal, denke ich. Wenn es irgendjemandem besser geht, tue es auf jeden Fall.

import h5py 

from keras.models import load_model 
from keras.models import save_model 


def load_model_ext(filepath, custom_objects=None): 
    model = load_model(filepath, custom_objects=None) 
    f = h5py.File(filepath, mode='r') 
    meta_data = None 
    if 'my_meta_data' in f.attrs: 
     meta_data = f.attrs.get('my_meta_data') 
    f.close() 
    return model, meta_data 


def save_model_ext(model, filepath, overwrite=True, meta_data=None): 
    save_model(model, filepath, overwrite) 
    if meta_data is not None: 
     f = h5py.File(filepath, mode='a') 
     f.attrs['my_meta_data'] = meta_data 
     f.close()