Skip to content

Instantly share code, notes, and snippets.

@talperetz
Created September 25, 2018 13:30
Show Gist options
  • Star 2 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
  • Save talperetz/6374ca542443535e76eb4139e2f5f21a to your computer and use it in GitHub Desktop.
Save talperetz/6374ca542443535e76eb4139e2f5f21a to your computer and use it in GitHub Desktop.
Catboost parameter space for hyperopt
import hyperopt
params_space = {
'learning_rate': hyperopt.hp.uniform('learning_rate', 0.01, 0.8),
'max_depth': scope.int(hyperopt.hp.quniform('max_depth', 2, 10, 1)),
'colsample_bylevel': hyperopt.hp.uniform('colsample_bylevel', 0.5, 1.0),
'bagging_temperature': hyperopt.hp.uniform('bagging_temperature', 0.0, 100),
'random_strength': hyperopt.hp.uniform('random_strength', 0.0, 100),
'scale_pos_weight': hyperopt.hp.uniform('scale_pos_weight', 1.0, 16.0), # change 16.0 to n_negative / n_poistive
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment