Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save omchaithanyav/cd768f065303fd49f3dc43865230c044 to your computer and use it in GitHub Desktop.
Save omchaithanyav/cd768f065303fd49f3dc43865230c044 to your computer and use it in GitHub Desktop.
from flaml import AutoML
# Initialize
automl = AutoML()
# Specify the constraints
automl_settings = {
"time_budget": 10, # in seconds
"metric": 'accuracy',
"task": 'classification',
"log_file_name": "iris.log",
}
X_train, y_train = X, y
automl.fit(X_train=X_train, y_train=y_train, **automl_settings)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment