ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix LightGBM example shows UserWarning

Browse Source
This commit is contained in:
allegroai 2022-07-01 11:37:42 +03:00
parent 1147931774
commit d1e8285760
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/frameworks/lightgbm/lightgbm_example.py
View File

@ -43,7 +43,8 @@ params = {
'feature_fraction': 0.9,
'bagging_fraction': 0.8,
'bagging_freq': 5,
'verbose': 0
'verbose': 0,
'force_col_wise': True,
}
print('Starting training...')
@ -53,8 +54,8 @@ gbm = lgb.train(
params,
lgb_train,
num_boost_round=20,
valid_sets=lgb_eval,
early_stopping_rounds=5
valid_sets=[lgb_eval],
callbacks=[lgb.early_stopping(stopping_rounds=5)],
)
print('Saving model...')