From da92238ffebc7c92ca6a4ba52c0cafaad8f532fb Mon Sep 17 00:00:00 2001 From: minghao Date: Tue, 1 Aug 2023 09:58:30 +0800 Subject: [PATCH] Commenting use_label_encoder - xgboost (#1122) * Commenting use_label_encoder - xgboost * format change * moving the import xgboost version to the head * Shfit params for use_label to outside maxdept * Keep the original logic --------- Co-authored-by: Shaokun Co-authored-by: Li Jiang --- flaml/automl/model.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/flaml/automl/model.py b/flaml/automl/model.py index d4970895e..6a0a0aa80 100644 --- a/flaml/automl/model.py +++ b/flaml/automl/model.py @@ -1508,7 +1508,12 @@ class XGBoostEstimator(SKLearnEstimator): params["grow_policy"] = params.get("grow_policy", "lossguide") params["tree_method"] = params.get("tree_method", "hist") # params["booster"] = params.get("booster", "gbtree") - params["use_label_encoder"] = params.get("use_label_encoder", False) + + # use_label_encoder is deprecated in 1.7. + from xgboost import __version__ as xgboost_version + + if xgboost_version < "1.7.0": + params["use_label_encoder"] = params.get("use_label_encoder", False) if "n_jobs" in config: params["nthread"] = params.pop("n_jobs") return params