Skip to content

GitLab

Unverified Commit 1d075f4d authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

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

* pass params to predictor

* hotfix
parent a7d2b174
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python-package/lightgbm/basic.py
View file @ 1d075f4d
...@@ -2123,8 +2123,8 @@ class Booster(object): ...@@ -2123,8 +2123,8 @@ class Booster(object):
result : Booster result : Booster
Refitted Booster. Refitted Booster.
""" """
predictor = self._to_predictor() predictor = self._to_predictor(kwargs)
leaf_preds = predictor.predict(data, -1, pred_leaf=True, **kwargs) leaf_preds = predictor.predict(data, -1, pred_leaf=True)
nrow, ncol = leaf_preds.shape nrow, ncol = leaf_preds.shape
train_set = Dataset(data, label, silent=True) train_set = Dataset(data, label, silent=True)
new_booster = Booster(self.params, train_set, silent=True) new_booster = Booster(self.params, train_set, silent=True)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment