Skip to content

Instantly share code, notes, and snippets.

@WillKoehrsen
Created July 18, 2018 00:03
Show Gist options
  • Save WillKoehrsen/1dfceb23b6569c3ca9b977c9b50a58d9 to your computer and use it in GitHub Desktop.
Save WillKoehrsen/1dfceb23b6569c3ca9b977c9b50a58d9 to your computer and use it in GitHub Desktop.
import numpy as np
# Hyperparameter grid
param_grid = {
'boosting_type': ['gbdt', 'goss', 'dart'],
'num_leaves': list(range(20, 150)),
'learning_rate': list(np.logspace(np.log10(0.005), np.log10(0.5), base = 10, num = 1000)),
'subsample_for_bin': list(range(20000, 300000, 20000)),
'min_child_samples': list(range(20, 500, 5)),
'reg_alpha': list(np.linspace(0, 1)),
'reg_lambda': list(np.linspace(0, 1)),
'colsample_bytree': list(np.linspace(0.6, 1, 10)),
'subsample': list(np.linspace(0.5, 1, 100)),
'is_unbalance': [True, False]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment