Skip to content

Instantly share code, notes, and snippets.

@WashingtonGold
Created September 11, 2020 16:48
Show Gist options
  • Save WashingtonGold/dec1264506c945a62d46e76fe00a8212 to your computer and use it in GitHub Desktop.
Save WashingtonGold/dec1264506c945a62d46e76fe00a8212 to your computer and use it in GitHub Desktop.
import autosklearn as ask
#ask.regression.AutoSklearnRegressor() for regression tasks
model = ask.classification.AutoSklearnClassifier(ensemble_size=10, #size of the end ensemble (minimum is 1)
time_left_for_this_task=120, #the number of seconds the process runs for
per_run_time_limit=30) #maximum seconds allocated per model
model.fit(X_train, y_train) #begin fitting the search model
print(model.sprint_statistics()) #print statistics for the search
y_predictions = model.predict(X_test) #get predictions from the model
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment