"tests/vscode:/vscode.git/clone" did not exist on "7f7bc44775d61367ce827ec4215e2a9af7b50747"
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):
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)
......
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