diff options
Diffstat (limited to 'lib/functions.py')
| -rw-r--r-- | lib/functions.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/functions.py b/lib/functions.py index 417c8c8..efed2f6 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -1054,6 +1054,7 @@ class LightGBMFunction(SKLearnRegressionFunction): import dfatool.lightgbm as lightgbm + lightgbm.register_logger(logger) lgbr = lightgbm.LGBMRegressor( boosting_type=boosting_type, n_estimators=n_estimators, |