Support learning rate for zero-hessian objectives. (#8866)

2023-03-06 20:33:28 +08:00
parent 173096a6a7
commit 228a46e8ad
34 changed files with 464 additions and 434 deletions
--- a/tests/python/test_pickling.py
+++ b/tests/python/test_pickling.py
@@ -51,11 +51,8 @@ class TestPickling:

    def test_model_pickling_json(self):
        def check(config):
-            updater = config["learner"]["gradient_booster"]["updater"]
-            if params["tree_method"] == "exact":
-                subsample = updater["grow_colmaker"]["train_param"]["subsample"]
-            else:
-                subsample = updater["grow_quantile_histmaker"]["train_param"]["subsample"]
+            tree_param = config["learner"]["gradient_booster"]["tree_train_param"]
+            subsample = tree_param["subsample"]
            assert float(subsample) == 0.5

        params = {"nthread": 8, "tree_method": "hist", "subsample": 0.5}