ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +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, 'feature_fraction': 0.9,
'bagging_fraction': 0.8, 'bagging_fraction': 0.8,
'bagging_freq': 5, 'bagging_freq': 5,
'verbose': 0 'verbose': 0,
'force_col_wise': True,
} }
print('Starting training...') print('Starting training...')
@ -53,8 +54,8 @@ gbm = lgb.train(
params, params,
lgb_train, lgb_train,
num_boost_round=20, num_boost_round=20,
valid_sets=lgb_eval, valid_sets=[lgb_eval],
early_stopping_rounds=5 callbacks=[lgb.early_stopping(stopping_rounds=5)],
) )
print('Saving model...') print('Saving model...')