Skip to content

Commit eadd3e1

Browse files
authored
Remove warning from lightGBM test. (#950)
``` [LightGBM] [Warning] Auto-choosing row-wise multi-threading, the overhead of testing was 0.000252 seconds. You can set `force_row_wise=true` to remove the overhead. ```
1 parent 4e20d1d commit eadd3e1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_lightgbm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ def test_cpu(self):
2121
'feature_fraction': 0.9,
2222
'bagging_fraction': 0.8,
2323
'bagging_freq': 5,
24+
'force_row_wise': True,
2425
'verbose': 0
2526
}
2627

@@ -46,6 +47,7 @@ def test_gpu(self):
4647
'feature_fraction': 0.9,
4748
'bagging_fraction': 0.8,
4849
'bagging_freq': 5,
50+
'force_row_wise': True,
4951
'verbose': 1,
5052
'device': 'gpu'
5153
}

0 commit comments

Comments
 (0)