Skip to content

GitLab

"git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "e757300f764d69b44ed81ec0d770257de5a050e2"
Unverified Commit 1bd15b94 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix refit decay rate in python (#2254)

parent bf78008b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
python-package/lightgbm/basic.py
View file @ 1bd15b94
......@@ -2362,7 +2362,9 @@ class Booster(object):
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)
new_params = copy.deepcopy(self.params)
new_params['refit_decay_rate'] = decay_rate
new_booster = Booster(new_params, train_set, silent=True)
# Copy models
_safe_call(_LIB.LGBM_BoosterMerge(
new_booster.handle,
......
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