diff options
| author | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-03-07 09:34:32 +0100 |
|---|---|---|
| committer | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-03-07 09:34:32 +0100 |
| commit | 1dfb6da4f8f1fec01d82f5f2dcbc66751d4078ec (patch) | |
| tree | 0f59e2a3263704dae9bad891f170258b65935066 | |
| parent | 29d10d5dd6c08bcafc7c34c48b8db599fcbd7e49 (diff) | |
lightgbm: use same logger as dfatool
| -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, |