[python] pass params to predictor in refit (#1650)

* pass params to predictor * hotfix

[python] pass params to predictor in refit (#1650)
* pass params to predictor * hotfix
1d075f4d · Nikita Titov · GitHub · a7d2b174 · 1d075f4d
Unverified Commit 1d075f4d authored Sep 08, 2018 by Nikita Titov Committed by GitHub Sep 08, 2018
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python-package/lightgbm/basic.py python-package/lightgbm/basic.py +2 -2

No files found.
--- a/python-package/lightgbm/basic.py
+++ b/python-package/lightgbm/basic.py
@@ -2123,8 +2123,8 @@ class Booster(object):
        result : Booster
            Refitted Booster.
        """
-        predictor = self._to_predictor()
-        leaf_preds = predictor.predict(data, -1, pred_leaf=True, **kwargs)
+        predictor = self._to_predictor(kwargs)
+        leaf_preds = predictor.predict(data, -1, pred_leaf=True)
        nrow, ncol = leaf_preds.shape
        train_set = Dataset(data, label, silent=True)
        new_booster = Booster(self.params, train_set, silent=True)