From 832b159f5ec22482d862a15422b76e0f28231433 Mon Sep 17 00:00:00 2001 From: jokedoke Date: Fri, 9 Jan 2026 15:17:08 +0300 Subject: [PATCH] Catboost rsm support and pruning callback mod (#36) --- quickadapter/user_data/strategies/Utils.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/strategies/Utils.py b/quickadapter/user_data/strategies/Utils.py index a2d4354..65d87ff 100644 --- a/quickadapter/user_data/strategies/Utils.py +++ b/quickadapter/user_data/strategies/Utils.py @@ -1878,6 +1878,7 @@ def fit_regressor( if task_type == "GPU": model_training_parameters.setdefault("max_ctr_complexity", 4) model_training_parameters.pop("n_jobs", None) + model_training_parameters.pop("rsm", None) else: n_jobs = model_training_parameters.pop("n_jobs", None) if n_jobs is not None: @@ -1902,8 +1903,14 @@ def fit_regressor( ) pruning_callback = None - if trial is not None and has_eval_set: - pruning_callback = optuna.integration.CatBoostPruningCallback(trial, "RMSE") + if ( + trial is not None + and has_eval_set + and task_type != "GPU" + ): + pruning_callback = optuna.integration.CatBoostPruningCallback( + trial, "RMSE" + ) fit_callbacks.append(pruning_callback) model = CatBoostRegressor(**model_training_parameters) -- 2.43.0