autogluon.tabular 1.3.2b20250714__tar.gz → 1.3.2b20250715__tar.gz

{autogluon.tabular-1.3.2b20250714 → autogluon.tabular-1.3.2b20250715}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.3.2b20250714
+Version: 1.3.2b20250715
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.tabular-1.3.2b20250714 → autogluon.tabular-1.3.2b20250715}/src/autogluon/tabular/models/catboost/catboost_model.py

@@ -13,13 +13,13 @@ from autogluon.common.features.types import R_BOOL, R_CATEGORY, R_FLOAT, R_INT
 from autogluon.common.utils.pandas_utils import get_approximate_df_mem_usage
 from autogluon.common.utils.resource_utils import ResourceManager
 from autogluon.common.utils.try_import import try_import_catboost
-from autogluon.core.constants import MULTICLASS, PROBLEM_TYPES_CLASSIFICATION, QUANTILE, SOFTCLASS
+from autogluon.core.constants import MULTICLASS, PROBLEM_TYPES_CLASSIFICATION, REGRESSION, QUANTILE, SOFTCLASS
 from autogluon.core.models import AbstractModel
 from autogluon.core.models._utils import get_early_stopping_rounds
 from autogluon.core.utils.exceptions import TimeLimitExceeded
 
 from .callbacks import EarlyStoppingCallback, MemoryCheckCallback, TimeCheckCallback
-from .catboost_utils import get_catboost_metric_from_ag_metric
+from .catboost_utils import get_catboost_metric_from_ag_metric, CATBOOST_EVAL_METRIC_TO_LOSS_FUNCTION
 from .hyperparameters.parameters import get_param_baseline
 from .hyperparameters.searchspaces import get_default_searchspace
 
@@ -131,11 +131,14 @@ class CatBoostModel(AbstractModel):
             # FIXME: This is extremely slow due to unoptimized metric / objective sent to CatBoost
             from .catboost_softclass_utils import SoftclassCustomMetric, SoftclassObjective
 
-            params["loss_function"] = SoftclassObjective.SoftLogLossObjective()
+            params.setdefault("loss_function",  SoftclassObjective.SoftLogLossObjective())
             params["eval_metric"] = SoftclassCustomMetric.SoftLogLossMetric()
-        elif self.problem_type == QUANTILE:
-            # FIXME: Unless specified, CatBoost defaults to loss_function='MultiQuantile' and raises an exception
-            params["loss_function"] = params["eval_metric"]
+        elif self.problem_type in [REGRESSION, QUANTILE]:
+            # Choose appropriate loss_function that is as close as possible to the eval_metric
+            params.setdefault(
+                "loss_function",
+                CATBOOST_EVAL_METRIC_TO_LOSS_FUNCTION.get(params["eval_metric"], params["eval_metric"])
+            )
 
         model_type = CatBoostClassifier if self.problem_type in PROBLEM_TYPES_CLASSIFICATION else CatBoostRegressor
         num_rows_train = len(X)

{autogluon.tabular-1.3.2b20250714 → autogluon.tabular-1.3.2b20250715}/src/autogluon/tabular/models/catboost/catboost_utils.py

@@ -6,6 +6,13 @@ logger = logging.getLogger(__name__)
 
 
 CATBOOST_QUANTILE_PREFIX = "Quantile:"
+# Mapping from non-optimizable eval_metric to optimizable loss_function.
+# See https://catboost.ai/docs/en/concepts/loss-functions-regression#usage-information
+CATBOOST_EVAL_METRIC_TO_LOSS_FUNCTION = {
+    "MedianAbsoluteError": "MAE",
+    "SMAPE": "MAPE",
+    "R2": "RMSE",
+}
 
 
 # TODO: Add weight support?
@@ -65,7 +72,10 @@ def get_catboost_metric_from_ag_metric(metric, problem_type, quantile_levels=Non
             mean_squared_error="RMSE",
             root_mean_squared_error="RMSE",
             mean_absolute_error="MAE",
+            mean_absolute_percentage_error="MAPE",
+            # Non-optimizable metrics, see CATBOOST_EVAL_METRIC_TO_LOSS_FUNCTION
             median_absolute_error="MedianAbsoluteError",
+            symmetric_mean_absolute_percentage_error="SMAPE",
             r2="R2",
         )
         metric_class = metric_map.get(metric.name, "RMSE")

{autogluon.tabular-1.3.2b20250714 → autogluon.tabular-1.3.2b20250715}/src/autogluon/tabular/models/lgb/lgb_model.py

@@ -281,7 +281,8 @@ class LGBModel(AbstractModel):
                 train_params["params"]["metric"] = f'{stopping_metric},{train_params["params"]["metric"]}'
 
         if self.problem_type == SOFTCLASS:
-            train_params["fobj"] = lgb_utils.softclass_lgbobj
+            train_params["params"]["objective"] = lgb_utils.softclass_lgbobj
+            train_params["params"]["num_classes"] = self.num_classes
         elif self.problem_type == QUANTILE:
             train_params["params"]["quantile_levels"] = self.quantile_levels
         if seed_val is not None:

{autogluon.tabular-1.3.2b20250714 → autogluon.tabular-1.3.2b20250715}/src/autogluon/tabular/testing/fit_helper.py

@@ -441,9 +441,9 @@ class FitHelper:
                 num_bag_sets=1,
             )
             if isinstance(bag, bool):
-                problem_types_bag = supported_problem_types
+                problem_types_bag = problem_types_to_check
             elif bag == "first":
-                problem_types_bag = supported_problem_types[:1]
+                problem_types_bag = problem_types_to_check[:1]
             else:
                 raise ValueError(f"Unknown 'bag' value: {bag}")
 

{autogluon.tabular-1.3.2b20250714 → autogluon.tabular-1.3.2b20250715}/src/autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.3.2b20250714"
+__version__ = "1.3.2b20250715"
 __lite__ = False

{autogluon.tabular-1.3.2b20250714 → autogluon.tabular-1.3.2b20250715}/src/autogluon.tabular.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.3.2b20250714
+Version: 1.3.2b20250715
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community

{autogluon.tabular-1.3.2b20250714 → autogluon.tabular-1.3.2b20250715}/src/autogluon.tabular.egg-info/requires.txt

@@ -3,21 +3,21 @@ scipy<1.17,>=1.5.4
 pandas<2.4.0,>=2.0.0
 scikit-learn<1.8.0,>=1.4.0
 networkx<4,>=3.0
-autogluon.core==1.3.2b20250714
-autogluon.features==1.3.2b20250714
+autogluon.core==1.3.2b20250715
+autogluon.features==1.3.2b20250715
 
 [all]
 spacy<3.9
-huggingface_hub[torch]
-fastai<2.9,>=2.3.1
-numpy<2.3.0,>=1.25
-pytabkit<1.6,>=1.5
-einops<0.9,>=0.7
-autogluon.core[all]==1.3.2b20250714
+torch<2.8,>=2.2
 xgboost<3.1,>=2.0
+numpy<2.3.0,>=1.25
 lightgbm<4.7,>=4.0
-torch<2.8,>=2.2
+huggingface_hub[torch]
+autogluon.core[all]==1.3.2b20250715
 catboost<1.3,>=1.2
+fastai<2.9,>=2.3.1
+einops<0.9,>=0.7
+pytabkit<1.6,>=1.5
 
 [catboost]
 numpy<2.3.0,>=1.25
@@ -41,7 +41,7 @@ omegaconf
 transformers
 
 [ray]
-autogluon.core[all]==1.3.2b20250714
+autogluon.core[all]==1.3.2b20250715
 
 [realmlp]
 pytabkit<1.6,>=1.5