[SKL] Propagate parameters to booster during set_param. (#6416)

2020-11-20 20:37:35 +08:00
parent cc581b3b6b
commit 2ce2a1a4d8
2 changed files with 21 additions and 1 deletions
--- a/python-package/xgboost/sklearn.py
+++ b/python-package/xgboost/sklearn.py
@@ -334,6 +334,10 @@ class XGBModel(XGBModelBase):
            else:
                self.kwargs[key] = value

+        if hasattr(self, '_Booster'):
+            parameters = self.get_xgb_params()
+            self.get_booster().set_param(parameters)
+
        return self

    def get_params(self, deep=True):