Skip to content

Instantly share code, notes, and snippets.

@jrzaurin
Last active March 10, 2019 10:15
Show Gist options
  • Save jrzaurin/a4cb3c3b53983eccb925c9e8a4a70844 to your computer and use it in GitHub Desktop.
Save jrzaurin/a4cb3c3b53983eccb925c9e8a4a70844 to your computer and use it in GitHub Desktop.
params = dict(
is_unbalance = True,
learning_rate = Real(0.01, 0.3),
num_boost_round=Integer(50, 500),
num_leaves=Integer(31, 255),
min_child_weight = Real(0.1, 10),
colsample_bytree= Real(0.5, 1.),
subsample=Real(0.5, 1.),
reg_alpha= Real(0.01, 0.1),
reg_lambda= Real(0.01, 0.1)
)
extra_params = dict(
early_stopping_rounds=20,
feature_name=colnames,
categorical_feature=categorical_columns
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment