Untitled

 avatar
unknown
python
10 months ago
647 B
2
Indexable
def entrenar_y_guardar_modelo():
    modelo, tokenizer, label_encoder = entrenar_modelo()    
    nombre_modelo = simpledialog.askstring("Guardar modelo", "Ingrese el nombre del archivo para guardar el modelo:")  
    save_model(modelo, f"{nombre_modelo}_modelo.h5")        
    with open(f"{nombre_modelo}_tokenizer.pickle", 'wb') as handle:             
        pickle.dump(tokenizer, handle, protocol=pickle.HIGHEST_PROTOCOL)        
    with open(f"{nombre_modelo}_label_encoder.pickle", 'wb') as handle:         
        pickle.dump(label_encoder, handle, protocol=pickle.HIGHEST_PROTOCOL)
    return modelo, tokenizer, label_encoder 
Editor is loading...
Leave a Comment