Gradient boosting - Adapt tests to version xgboost=2.0.0

biolab · Sep 12, 2023 · 5ce3c81 · 5ce3c81
1 parent 24596fe
commit 5ce3c81
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 10 deletions.
diff --git a/Orange/widgets/model/tests/test_owgradientboosting.py b/Orange/widgets/model/tests/test_owgradientboosting.py
@@ -159,8 +159,7 @@ def test_default_parameters_cls(self):
         model = booster(data)
         params = model.skl_model.get_params()
         booster_params = json.loads(model.skl_model.get_booster().save_config())
-        updater = booster_params["learner"]["gradient_booster"]["updater"]
-        tp = updater["grow_colmaker"]["train_param"]
+        tp = booster_params["learner"]["gradient_booster"]["tree_train_param"]
         self.assertEqual(params["n_estimators"], self.editor.n_estimators)
         self.assertEqual(
             round(float(tp["learning_rate"]), 1), self.editor.learning_rate
@@ -179,8 +178,7 @@ def test_default_parameters_reg(self):
         model = booster(data)
         params = model.skl_model.get_params()
         booster_params = json.loads(model.skl_model.get_booster().save_config())
-        updater = booster_params["learner"]["gradient_booster"]["updater"]
-        tp = updater["grow_colmaker"]["train_param"]
+        tp = booster_params["learner"]["gradient_booster"]["tree_train_param"]
         self.assertEqual(params["n_estimators"], self.editor.n_estimators)
         self.assertEqual(
             round(float(tp["learning_rate"]), 1), self.editor.learning_rate
@@ -224,8 +222,7 @@ def test_default_parameters_cls(self):
         model = booster(data)
         params = model.skl_model.get_params()
         booster_params = json.loads(model.skl_model.get_booster().save_config())
-        updater = booster_params["learner"]["gradient_booster"]["updater"]
-        tp = updater["grow_colmaker"]["train_param"]
+        tp = booster_params["learner"]["gradient_booster"]["tree_train_param"]
         self.assertEqual(params["n_estimators"], self.editor.n_estimators)
         self.assertEqual(
             round(float(tp["learning_rate"]), 1), self.editor.learning_rate
@@ -244,8 +241,7 @@ def test_default_parameters_reg(self):
         model = booster(data)
         params = model.skl_model.get_params()
         booster_params = json.loads(model.skl_model.get_booster().save_config())
-        updater = booster_params["learner"]["gradient_booster"]["updater"]
-        tp = updater["grow_colmaker"]["train_param"]
+        tp = booster_params["learner"]["gradient_booster"]["tree_train_param"]
         self.assertEqual(params["n_estimators"], self.editor.n_estimators)
         self.assertEqual(
             round(float(tp["learning_rate"]), 1), self.editor.learning_rate

diff --git a/requirements-opt.txt b/requirements-opt.txt
@@ -1,2 +1,2 @@
 catboost>=1.0.1
-xgboost>=1.5.0
+xgboost>=2.0.0
diff --git a/tox.ini b/tox.ini
@@ -66,7 +66,7 @@ deps =
     # oldest: pyyaml
     # oldest: openpyxl
     oldest: httpx==0.21.0
-    oldest: xgboost==1.5.0
+    oldest: xgboost==2.0.0
     oldest: catboost==1.0.1
 
 commands_pre =