Skip to content

Instantly share code, notes, and snippets.

@talperetz

talperetz/catboost_skopt_params.py

Last active November 1, 2018 00:33
Show Gist options
  • Save talperetz/7049dbd90956feadc8978bc34978f8e5 to your computer and use it in GitHub Desktop.
Save talperetz/7049dbd90956feadc8978bc34978f8e5 to your computer and use it in GitHub Desktop.
Download ZIP
Catboost parameter space for skopt
Raw
catboost_skopt_params.py
from skopt.space import Real, Integer
params_space = [Real(0.01, 0.8, name='learning_rate'),
Integer(2, 10, name='max_depth'),
Real(0.5, 1.0, name='colsample_bylevel'),
Real(1.0, 16.0, name='scale_pos_weight'), # change 16.0 to n_negative / n_poistive
Real(0.0, 100, name='bagging_temperature'),
Real(0.0, 100, name='random_strength'),
Real(0.0, 100, name='reg_lambda')]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment