"...git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "a2c1dec0722c3cfe614d6cd7718b306216053693"
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
...@@ -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