Skip to content

Instantly share code, notes, and snippets.

@alexeyev
Created April 22, 2020 12:36
Show Gist options
  • Save alexeyev/c72acb9b98c44368c54993e02dbf215a to your computer and use it in GitHub Desktop.
Save alexeyev/c72acb9b98c44368c54993e02dbf215a to your computer and use it in GitHub Desktop.
# import autosklearn
# from autosklearn.classification import AutoSklearnClassifier
# # print("Available CLASSIFICATION metrics autosklearn.metrics.*:")
# # print("\t*" + "\n\t*".join(autosklearn.metrics.CLASSIFICATION_METRICS))
# print("Y labels:", set(y))
#
# automl = autosklearn.classification.AutoSklearnClassifier(
# n_jobs=3,
# ensemble_size=1,
# seed=100,
# time_left_for_this_task=600,
# resampling_strategy=StratifiedKFold,
# resampling_strategy_arguments={"n_splits": 3, "shuffle": True},
# # exclude_estimators=["extra_trees"],
# include_estimators=["random_forest"],
# )
# automl.fit(X, y, dataset_name='ru_pgg', metric=autosklearn.metrics.f1_macro)
# print(automl.sprint_statistics())
# print(automl.show_models())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment