keras-team/autokeras

Can the models not be saved during the trial?

brunovaldebenito opened this issue · 0 comments

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