Skip to content

Instantly share code, notes, and snippets.

@dmarcelinobr

dmarcelinobr/model_tuning.R

Created August 7, 2020 22:46
Show Gist options
  • Save dmarcelinobr/990262c9108cb02188f3815d3d52eecd to your computer and use it in GitHub Desktop.
Save dmarcelinobr/990262c9108cb02188f3815d3d52eecd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_tuning.R
n_random = 100
params_grid <- expand.grid(
num_leaves = c(100, 200, 500, 1000, 2000, 5000),
learning_rate = c(0.01, 0.03, 0.1),
min_data_in_leaf = c(5, 10,20, 50),
feature_fraction = c(0.6, 0.8, 1),
bagging_fraction = c(0.4, 0.6, 0.8, 1),
lambda_l1 = c(0.0, 0.0, 0.01, 0.1, 0.3),
lambda_l2 = c(0.0, 0.0, 0.01, 0.1, 0.3)
# TODO
# min_sum_hessian_in_leaf
# min_gain_to_split
# max_bin
# min_data_in_bin
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment