Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
brunovaldebenito opened this issue Apr 13, 2023 · 0 comments
Open

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

brunovaldebenito opened this issue Apr 13, 2023 · 0 comments

Comments

@brunovaldebenito
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant