keras-team / autokeras

AutoML library for deep learning
http://autokeras.com/
Apache License 2.0
9.1k stars 1.4k forks source link

Can the models not be saved during the trial? #1877

Open brunovaldebenito opened 1 year ago

brunovaldebenito commented 1 year ago

When you build an autokeras model, and train it with 'fit', it generates a folder with the name of the project in which all the trials are saved. I would like to know if there is a parameter that avoids saving all of them, and only saves the best one.

clf = ak.StructuredDataClassifier(overwrite=True, max_trials=3)
history= clf.fit(x_train, y_train, verbose=0, epochs=25)

It saves in 'structured_data_clasifier' 3 models: 'trial_00', 'trial_01', 'trial_02'. how do I save only the best model, either by saving this part and using only the model.save, or some other way?

thanks