Skip to content

Instantly share code, notes, and snippets.

@shashankprasanna
Created February 12, 2020 00:20
Show Gist options
  • Save shashankprasanna/6e306df8d81840cac73efba1a6f411ee to your computer and use it in GitHub Desktop.
Save shashankprasanna/6e306df8d81840cac73efba1a6f411ee to your computer and use it in GitHub Desktop.
from sagemaker.tuner import IntegerParameter, CategoricalParameter, ContinuousParameter, HyperparameterTuner
hyperparameter_ranges = {
'n_estimators' : IntegerParameter(10, 200),
'max_depth' : IntegerParameter(1, 22),
'n_bins' : IntegerParameter(5, 24),
'split_criterion' : CategoricalParameter([0, 1]),
'split_algo' : CategoricalParameter([0, 1]),
'bootstrap' : CategoricalParameter([True, False]),
'bootstrap_features' : CategoricalParameter([True, False]),
'max_features' : ContinuousParameter(0.01, 0.5),
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment