autogluon.tabular 1.4.1b20251201__tar.gz → 1.5.1b20260114__tar.gz

{autogluon_tabular-1.4.1b20251201/src/autogluon.tabular.egg-info → autogluon_tabular-1.5.1b20260114}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: autogluon.tabular
-Version: 1.4.1b20251201
+Version: 1.5.1b20260114
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -40,101 +40,102 @@ Requires-Dist: scipy<1.17,>=1.5.4
 Requires-Dist: pandas<2.4.0,>=2.0.0
 Requires-Dist: scikit-learn<1.8.0,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.4.1b20251201
-Requires-Dist: autogluon.features==1.4.1b20251201
+Requires-Dist: autogluon.core==1.5.1b20260114
+Requires-Dist: autogluon.features==1.5.1b20260114
 Provides-Extra: lightgbm
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "lightgbm"
 Provides-Extra: catboost
-Requires-Dist: numpy<2.3.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
 Provides-Extra: xgboost
-Requires-Dist: xgboost<3.1,>=2.0; extra == "xgboost"
+Requires-Dist: xgboost<3.2,>=2.0; extra == "xgboost"
 Provides-Extra: realmlp
-Requires-Dist: pytabkit<1.7,>=1.6; extra == "realmlp"
+Requires-Dist: pytabkit<1.8,>=1.7.2; extra == "realmlp"
 Provides-Extra: interpret
 Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "interpret"
 Provides-Extra: fastai
 Requires-Dist: spacy<3.9; extra == "fastai"
-Requires-Dist: torch<2.8,>=2.6; extra == "fastai"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "fastai"
+Requires-Dist: torch<2.10,>=2.6; extra == "fastai"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "fastai"
 Provides-Extra: tabm
-Requires-Dist: torch<2.8,>=2.6; extra == "tabm"
+Requires-Dist: torch<2.10,>=2.6; extra == "tabm"
 Provides-Extra: tabpfn
-Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabpfn"
+Requires-Dist: tabpfn<6.2.1,>=6.2.0; extra == "tabpfn"
+Provides-Extra: tabdpt
+Requires-Dist: tabdpt<1.2,>=1.1.11; extra == "tabdpt"
 Provides-Extra: tabpfnmix
-Requires-Dist: torch<2.8,>=2.6; extra == "tabpfnmix"
+Requires-Dist: torch<2.10,>=2.6; extra == "tabpfnmix"
 Requires-Dist: huggingface_hub[torch]<1.0; extra == "tabpfnmix"
 Requires-Dist: einops<0.9,>=0.7; extra == "tabpfnmix"
 Provides-Extra: mitra
 Requires-Dist: loguru; extra == "mitra"
 Requires-Dist: einx; extra == "mitra"
 Requires-Dist: omegaconf; extra == "mitra"
-Requires-Dist: torch<2.8,>=2.6; extra == "mitra"
+Requires-Dist: torch<2.10,>=2.6; extra == "mitra"
 Requires-Dist: transformers; extra == "mitra"
 Requires-Dist: huggingface_hub[torch]<1.0; extra == "mitra"
 Requires-Dist: einops<0.9,>=0.7; extra == "mitra"
 Provides-Extra: tabicl
-Requires-Dist: tabicl<0.2,>=0.1.3; extra == "tabicl"
+Requires-Dist: tabicl<0.2,>=0.1.4; extra == "tabicl"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.4.1b20251201; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.5.1b20260114; extra == "ray"
 Provides-Extra: skex
-Requires-Dist: scikit-learn-intelex<2025.5,>=2024.0; extra == "skex"
+Requires-Dist: scikit-learn-intelex<2025.10,>=2025.0; extra == "skex"
 Provides-Extra: imodels
 Requires-Dist: imodels<2.1.0,>=1.3.10; extra == "imodels"
 Provides-Extra: skl2onnx
-Requires-Dist: onnx<1.16.2,>=1.13.0; platform_system == "Windows" and extra == "skl2onnx"
-Requires-Dist: onnx<1.18.0,>=1.13.0; platform_system != "Windows" and extra == "skl2onnx"
-Requires-Dist: skl2onnx<1.18.0,>=1.15.0; extra == "skl2onnx"
-Requires-Dist: onnxruntime<1.20.0,>=1.17.0; extra == "skl2onnx"
-Requires-Dist: onnxruntime-gpu<1.20.0,>=1.17.0; extra == "skl2onnx"
+Requires-Dist: skl2onnx<1.20.0,>=1.15.0; extra == "skl2onnx"
+Requires-Dist: onnx!=1.16.2,<1.21.0,>=1.13.0; platform_system == "Windows" and extra == "skl2onnx"
+Requires-Dist: onnx<1.21.0,>=1.13.0; platform_system != "Windows" and extra == "skl2onnx"
+Requires-Dist: onnxruntime<1.24.0,>=1.17.0; extra == "skl2onnx"
+Requires-Dist: onnxruntime-gpu<1.24.0,>=1.17.0; (platform_system != "Darwin" and platform_machine != "aarch64") and extra == "skl2onnx"
 Provides-Extra: all
-Requires-Dist: spacy<3.9; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
-Requires-Dist: einx; extra == "all"
-Requires-Dist: loguru; extra == "all"
-Requires-Dist: transformers; extra == "all"
-Requires-Dist: autogluon.core[all]==1.4.1b20251201; extra == "all"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: omegaconf; extra == "all"
-Requires-Dist: huggingface_hub[torch]<1.0; extra == "all"
+Requires-Dist: torch<2.10,>=2.6; extra == "all"
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
-Requires-Dist: torch<2.8,>=2.6; extra == "all"
-Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
+Requires-Dist: loguru; extra == "all"
+Requires-Dist: einx; extra == "all"
+Requires-Dist: huggingface_hub[torch]<1.0; extra == "all"
+Requires-Dist: autogluon.core[all]==1.5.1b20260114; extra == "all"
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: transformers; extra == "all"
+Requires-Dist: omegaconf; extra == "all"
+Requires-Dist: xgboost<3.2,>=2.0; extra == "all"
+Requires-Dist: spacy<3.9; extra == "all"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "all"
 Provides-Extra: tabarena
-Requires-Dist: spacy<3.9; extra == "tabarena"
+Requires-Dist: einops<0.9,>=0.7; extra == "tabarena"
+Requires-Dist: torch<2.10,>=2.6; extra == "tabarena"
+Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
 Requires-Dist: einx; extra == "tabarena"
-Requires-Dist: transformers; extra == "tabarena"
 Requires-Dist: loguru; extra == "tabarena"
-Requires-Dist: autogluon.core[all]==1.4.1b20251201; extra == "tabarena"
-Requires-Dist: pytabkit<1.7,>=1.6; extra == "tabarena"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "tabarena"
-Requires-Dist: einops<0.9,>=0.7; extra == "tabarena"
+Requires-Dist: huggingface_hub[torch]<1.0; extra == "tabarena"
+Requires-Dist: tabpfn<6.2.1,>=6.2.0; extra == "tabarena"
+Requires-Dist: autogluon.core[all]==1.5.1b20260114; extra == "tabarena"
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
-Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "tabarena"
+Requires-Dist: transformers; extra == "tabarena"
 Requires-Dist: omegaconf; extra == "tabarena"
-Requires-Dist: huggingface_hub[torch]<1.0; extra == "tabarena"
-Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
-Requires-Dist: tabicl<0.2,>=0.1.3; extra == "tabarena"
-Requires-Dist: torch<2.8,>=2.6; extra == "tabarena"
-Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabarena"
-Requires-Dist: numpy<2.3.0,>=1.25; extra == "tabarena"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
+Requires-Dist: tabicl<0.2,>=0.1.4; extra == "tabarena"
+Requires-Dist: tabdpt<1.2,>=1.1.11; extra == "tabarena"
+Requires-Dist: xgboost<3.2,>=2.0; extra == "tabarena"
+Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "tabarena"
+Requires-Dist: pytabkit<1.8,>=1.7.2; extra == "tabarena"
+Requires-Dist: spacy<3.9; extra == "tabarena"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "tabarena"
 Provides-Extra: tests
 Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "tests"
-Requires-Dist: tabicl<0.2,>=0.1.3; extra == "tests"
-Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tests"
-Requires-Dist: pytabkit<1.7,>=1.6; extra == "tests"
-Requires-Dist: torch<2.8,>=2.6; extra == "tests"
+Requires-Dist: tabdpt<1.2,>=1.1.11; extra == "tests"
+Requires-Dist: tabicl<0.2,>=0.1.4; extra == "tests"
+Requires-Dist: tabpfn<6.2.1,>=6.2.0; extra == "tests"
+Requires-Dist: pytabkit<1.8,>=1.7.2; extra == "tests"
+Requires-Dist: torch<2.10,>=2.6; extra == "tests"
 Requires-Dist: huggingface_hub[torch]<1.0; extra == "tests"
 Requires-Dist: einops<0.9,>=0.7; extra == "tests"
 Requires-Dist: imodels<2.1.0,>=1.3.10; extra == "tests"
-Requires-Dist: onnx<1.16.2,>=1.13.0; platform_system == "Windows" and extra == "tests"
-Requires-Dist: onnx<1.18.0,>=1.13.0; platform_system != "Windows" and extra == "tests"
-Requires-Dist: skl2onnx<1.18.0,>=1.15.0; extra == "tests"
-Requires-Dist: onnxruntime<1.20.0,>=1.17.0; extra == "tests"
-Requires-Dist: onnxruntime-gpu<1.20.0,>=1.17.0; extra == "tests"
+Requires-Dist: skl2onnx<1.20.0,>=1.15.0; extra == "tests"
+Requires-Dist: onnx!=1.16.2,<1.21.0,>=1.13.0; platform_system == "Windows" and extra == "tests"
+Requires-Dist: onnx<1.21.0,>=1.13.0; platform_system != "Windows" and extra == "tests"
+Requires-Dist: onnxruntime<1.24.0,>=1.17.0; extra == "tests"
+Requires-Dist: onnxruntime-gpu<1.24.0,>=1.17.0; (platform_system != "Darwin" and platform_machine != "aarch64") and extra == "tests"
 Dynamic: author
 Dynamic: classifier
 Dynamic: description

{autogluon_tabular-1.4.1b20251201 → autogluon_tabular-1.5.1b20260114}/setup.py

@@ -41,14 +41,13 @@ extras_require = {
         "lightgbm>=4.0,<4.7",  # <{N+1} upper cap, where N is the latest released minor version
     ],
     "catboost": [
-        "numpy>=1.25,<2.3.0",
         "catboost>=1.2,<1.3",
     ],
     "xgboost": [
-        "xgboost>=2.0,<3.1",  # <{N+1} upper cap, where N is the latest released minor version
+        "xgboost>=2.0,<3.2",  # <{N+1} upper cap, where N is the latest released minor version
     ],
     "realmlp": [
-        "pytabkit>=1.6,<1.7",
+        "pytabkit>=1.7.2,<1.8",
     ],
     "interpret": [
         "interpret-core>=0.7.2,<0.8",
@@ -56,13 +55,16 @@ extras_require = {
     "fastai": [
         "spacy<3.9",
         "torch",  # version range defined in `core/_setup_utils.py`
-        "fastai>=2.3.1,<2.9",  # <{N+1} upper cap, where N is the latest released minor version
+        "fastai>=2.3.1,<2.8.6",  # Cap due to dependency conflict in fastai-2.8.6 https://github.com/autogluon/autogluon/issues/5521
     ],
     "tabm": [
         "torch",  # version range defined in `core/_setup_utils.py`
     ],
     "tabpfn": [
-        "tabpfn>=2.0.9,<2.2",  # <{N+1} upper cap, where N is the latest released minor version
+        "tabpfn>=6.2.0,<6.2.1",  # <{N+1} upper cap, where N is the latest released minor version
+    ],
+    "tabdpt": [
+        "tabdpt>=1.1.11,<1.2",
     ],
     "tabpfnmix": [
         "torch",  # version range defined in `core/_setup_utils.py`
@@ -79,41 +81,29 @@ extras_require = {
         "einops>=0.7,<0.9",
     ],
     "tabicl": [
-        "tabicl>=0.1.3,<0.2",  # 0.1.3 added a major bug fix to multithreading.
+        "tabicl>=0.1.4,<0.2",  # 0.1.4 added python 3.13 support
     ],
     "ray": [
         f"{ag.PACKAGE_NAME}.core[all]=={version}",
     ],
     "skex": [
-        "scikit-learn-intelex>=2024.0,<2025.5",  # <{N+1} upper cap, where N is the latest released minor version
+        "scikit-learn-intelex>=2025.0,<2025.10",  # <{N+1} upper cap, where N is the latest released minor version
     ],
     "imodels": [
         "imodels>=1.3.10,<2.1.0",  # 1.3.8/1.3.9 either remove/renamed attribute `complexity_` causing failures. https://github.com/csinva/imodels/issues/147
     ],
 }
 
-is_aarch64 = platform.machine() == "aarch64"
-is_darwin = sys.platform == "darwin"
-
-if is_darwin or is_aarch64:
-    # For macOS or aarch64, only use CPU version
-    extras_require["skl2onnx"] = [
-        "onnx>=1.13.0,<1.16.2;platform_system=='Windows'",  # cap at 1.16.1 for issue https://github.com/onnx/onnx/issues/6267
-        "onnx>=1.13.0,<1.18.0;platform_system!='Windows'",
-        "skl2onnx>=1.15.0,<1.18.0",
-        # For macOS, there isn't a onnxruntime-gpu package installed with skl2onnx.
-        # Therefore, we install onnxruntime explicitly here just for macOS.
-        "onnxruntime>=1.17.0,<1.20.0",
-    ]
-else:
-    # For other platforms, include both CPU and GPU versions
-    extras_require["skl2onnx"] = [
-        "onnx>=1.13.0,<1.16.2;platform_system=='Windows'",  # cap at 1.16.1 for issue https://github.com/onnx/onnx/issues/6267
-        "onnx>=1.13.0,<1.18.0;platform_system!='Windows'",
-        "skl2onnx>=1.15.0,<1.18.0",
-        "onnxruntime>=1.17.0,<1.20.0",  # install for gpu system due to https://github.com/autogluon/autogluon/issues/3804
-        "onnxruntime-gpu>=1.17.0,<1.20.0",
-    ]
+extras_require["skl2onnx"] = [
+    "skl2onnx>=1.15.0,<1.20.0",
+    # Sync ONNX requirements with multimodal/setup.py
+    "onnx>=1.13.0,!=1.16.2,<1.21.0;platform_system=='Windows'",  # exclude 1.16.2 for issue https://github.com/onnx/onnx/issues/6267
+    "onnx>=1.13.0,<1.21.0;platform_system!='Windows'",
+    # For macOS, there isn't a onnxruntime-gpu package installed with skl2onnx.
+    # Therefore, we install onnxruntime explicitly here just for macOS.
+    "onnxruntime>=1.17.0,<1.24.0",
+    "onnxruntime-gpu>=1.17.0,<1.24.0; platform_system != 'Darwin' and platform_machine != 'aarch64'",
+]
 
 # TODO: v1.0: Rename `all` to `core`, make `all` contain everything.
 all_requires = []
@@ -133,6 +123,7 @@ extras_require["all"] = all_requires
 tabarena_requires = copy.deepcopy(all_requires)
 for extra_package in [
     "interpret",
+    "tabdpt",
     "tabicl",
     "tabpfn",
     "realmlp",
@@ -144,6 +135,7 @@ extras_require["tabarena"] = tabarena_requires
 test_requires = []
 for test_package in [
     "interpret",
+    "tabdpt",
     "tabicl",  # Currently has unnecessary extra dependencies such as xgboost and wandb
     "tabpfn",
     "realmlp",  # Will consider to put as part of `all_requires` once part of a portfolio

{autogluon_tabular-1.4.1b20251201 → autogluon_tabular-1.5.1b20260114}/src/autogluon/tabular/__init__.py

@@ -1,5 +1,6 @@
 # noinspection PyUnresolvedReferences
 from autogluon.common.dataset import TabularDataset
+
 # noinspection PyUnresolvedReferences
 from autogluon.common.features.feature_metadata import FeatureMetadata
 from autogluon.common.utils.log_utils import _add_stream_handler

{autogluon_tabular-1.4.1b20251201 → autogluon_tabular-1.5.1b20260114}/src/autogluon/tabular/configs/config_helper.py

@@ -125,7 +125,9 @@ class ConfigBuilder:
 
         if isinstance(presets, list):
             unknown_keys = [k for k in presets if k not in valid_keys]
-            assert len(unknown_keys) == 0, f"The following presets are not recognized: {unknown_keys} - use one of the valid presets: {valid_keys}"
+            assert len(unknown_keys) == 0, (
+                f"The following presets are not recognized: {unknown_keys} - use one of the valid presets: {valid_keys}"
+            )
 
         self.config["presets"] = presets
         return self
@@ -144,12 +146,18 @@ class ConfigBuilder:
         valid_keys = self._valid_keys()
         valid_str_values = list(hyperparameter_config_dict.keys())
         if isinstance(hyperparameters, str):
-            assert hyperparameters in hyperparameter_config_dict, f"{hyperparameters} is not one of the valid presets {valid_str_values}"
+            assert hyperparameters in hyperparameter_config_dict, (
+                f"{hyperparameters} is not one of the valid presets {valid_str_values}"
+            )
         elif isinstance(hyperparameters, dict):
             unknown_keys = [k for k in hyperparameters.keys() if isinstance(k, str) and (k not in valid_keys)]
-            assert len(unknown_keys) == 0, f"The following model types are not recognized: {unknown_keys} - use one of the valid models: {valid_keys}"
+            assert len(unknown_keys) == 0, (
+                f"The following model types are not recognized: {unknown_keys} - use one of the valid models: {valid_keys}"
+            )
         else:
-            raise ValueError(f"hyperparameters must be either str: {valid_str_values} or dict with keys of {valid_keys}")
+            raise ValueError(
+                f"hyperparameters must be either str: {valid_str_values} or dict with keys of {valid_keys}"
+            )
         self.config["hyperparameters"] = hyperparameters
         return self
 
@@ -230,7 +238,9 @@ class ConfigBuilder:
         """
         valid_str_values = scheduler_factory._scheduler_presets.keys()
         if isinstance(hyperparameter_tune_kwargs, str):
-            assert hyperparameter_tune_kwargs in valid_str_values, f"{hyperparameter_tune_kwargs} string must be one of {valid_str_values}"
+            assert hyperparameter_tune_kwargs in valid_str_values, (
+                f"{hyperparameter_tune_kwargs} string must be one of {valid_str_values}"
+            )
         elif not isinstance(hyperparameter_tune_kwargs, dict):
             raise ValueError(f"hyperparameter_tune_kwargs must be either str: {valid_str_values} or dict")
         self.config["hyperparameter_tune_kwargs"] = hyperparameter_tune_kwargs
@@ -294,7 +304,9 @@ class ConfigBuilder:
             models = [models]
 
         unknown_keys = [k for k in models if isinstance(k, str) and (k not in valid_keys)]
-        assert len(unknown_keys) == 0, f"The following model types are not recognized: {unknown_keys} - use one of the valid models: {valid_keys}"
+        assert len(unknown_keys) == 0, (
+            f"The following model types are not recognized: {unknown_keys} - use one of the valid models: {valid_keys}"
+        )
 
         models = [m for m in valid_keys if m not in models]
         self.config["excluded_model_types"] = models

{autogluon_tabular-1.4.1b20251201 → autogluon_tabular-1.5.1b20260114}/src/autogluon/tabular/configs/feature_generator_presets.py

@@ -18,7 +18,9 @@ def get_default_feature_generator(feature_generator, feature_metadata=None, init
         elif feature_generator == "interpretable":
             feature_generator = AutoMLInterpretablePipelineFeatureGenerator(**init_kwargs)
         else:
-            raise ValueError(f"Unknown feature_generator preset: '{feature_generator}', valid presets: {['auto', 'interpretable']}")
+            raise ValueError(
+                f"Unknown feature_generator preset: '{feature_generator}', valid presets: {['auto', 'interpretable']}"
+            )
     if feature_metadata is not None:
         if feature_generator.feature_metadata_in is None and not feature_generator.is_fit():
             feature_generator.feature_metadata_in = copy.deepcopy(feature_metadata)

{autogluon_tabular-1.4.1b20251201 → autogluon_tabular-1.5.1b20260114}/src/autogluon/tabular/configs/hyperparameter_configs.py

@@ -2,6 +2,8 @@ import copy
 
 from .zeroshot.zeroshot_portfolio_2023 import hyperparameter_portfolio_zeroshot_2023
 from .zeroshot.zeroshot_portfolio_2025 import hyperparameter_portfolio_zeroshot_2025_small
+from .zeroshot.zeroshot_portfolio_cpu_2025_12_18 import hyperparameter_portfolio_zeroshot_cpu_2025_12_18
+from .zeroshot.zeroshot_portfolio_gpu_2025_12_18 import hyperparameter_portfolio_zeroshot_gpu_2025_12_18
 
 # Dictionary of preset hyperparameter configurations.
 hyperparameter_config_dict = dict(
@@ -25,12 +27,18 @@ hyperparameter_config_dict = dict(
         "RF": [
             {"criterion": "gini", "ag_args": {"name_suffix": "Gini", "problem_types": ["binary", "multiclass"]}},
             {"criterion": "entropy", "ag_args": {"name_suffix": "Entr", "problem_types": ["binary", "multiclass"]}},
-            {"criterion": "squared_error", "ag_args": {"name_suffix": "MSE", "problem_types": ["regression", "quantile"]}},
+            {
+                "criterion": "squared_error",
+                "ag_args": {"name_suffix": "MSE", "problem_types": ["regression", "quantile"]},
+            },
         ],
         "XT": [
             {"criterion": "gini", "ag_args": {"name_suffix": "Gini", "problem_types": ["binary", "multiclass"]}},
             {"criterion": "entropy", "ag_args": {"name_suffix": "Entr", "problem_types": ["binary", "multiclass"]}},
-            {"criterion": "squared_error", "ag_args": {"name_suffix": "MSE", "problem_types": ["regression", "quantile"]}},
+            {
+                "criterion": "squared_error",
+                "ag_args": {"name_suffix": "MSE", "problem_types": ["regression", "quantile"]},
+            },
         ],
     },
     # Results in smaller models. Generally will make inference speed much faster and disk usage much lower, but with worse accuracy.
@@ -51,14 +59,38 @@ hyperparameter_config_dict = dict(
         "XGB": {},
         "FASTAI": {},
         "RF": [
-            {"criterion": "gini", "max_depth": 15, "ag_args": {"name_suffix": "Gini", "problem_types": ["binary", "multiclass"]}},
-            {"criterion": "entropy", "max_depth": 15, "ag_args": {"name_suffix": "Entr", "problem_types": ["binary", "multiclass"]}},
-            {"criterion": "squared_error", "max_depth": 15, "ag_args": {"name_suffix": "MSE", "problem_types": ["regression", "quantile"]}},
+            {
+                "criterion": "gini",
+                "max_depth": 15,
+                "ag_args": {"name_suffix": "Gini", "problem_types": ["binary", "multiclass"]},
+            },
+            {
+                "criterion": "entropy",
+                "max_depth": 15,
+                "ag_args": {"name_suffix": "Entr", "problem_types": ["binary", "multiclass"]},
+            },
+            {
+                "criterion": "squared_error",
+                "max_depth": 15,
+                "ag_args": {"name_suffix": "MSE", "problem_types": ["regression", "quantile"]},
+            },
         ],
         "XT": [
-            {"criterion": "gini", "max_depth": 15, "ag_args": {"name_suffix": "Gini", "problem_types": ["binary", "multiclass"]}},
-            {"criterion": "entropy", "max_depth": 15, "ag_args": {"name_suffix": "Entr", "problem_types": ["binary", "multiclass"]}},
-            {"criterion": "squared_error", "max_depth": 15, "ag_args": {"name_suffix": "MSE", "problem_types": ["regression", "quantile"]}},
+            {
+                "criterion": "gini",
+                "max_depth": 15,
+                "ag_args": {"name_suffix": "Gini", "problem_types": ["binary", "multiclass"]},
+            },
+            {
+                "criterion": "entropy",
+                "max_depth": 15,
+                "ag_args": {"name_suffix": "Entr", "problem_types": ["binary", "multiclass"]},
+            },
+            {
+                "criterion": "squared_error",
+                "max_depth": 15,
+                "ag_args": {"name_suffix": "MSE", "problem_types": ["regression", "quantile"]},
+            },
         ],
     },
     # Results in much smaller models. Behaves similarly to 'light', but in many cases with over 10x less disk usage and a further reduction in accuracy.
@@ -117,6 +149,8 @@ hyperparameter_config_dict = dict(
     zeroshot=hyperparameter_portfolio_zeroshot_2023,
     zeroshot_2023=hyperparameter_portfolio_zeroshot_2023,
     zeroshot_2025_tabfm=hyperparameter_portfolio_zeroshot_2025_small,
+    zeroshot_2025_12_18_gpu=hyperparameter_portfolio_zeroshot_gpu_2025_12_18,
+    zeroshot_2025_12_18_cpu=hyperparameter_portfolio_zeroshot_cpu_2025_12_18,
 )
 
 tabpfnmix_default = {
@@ -133,6 +167,7 @@ hyperparameter_config_dict["experimental_2024"] = {"TABPFNMIX": tabpfnmix_defaul
 hyperparameter_config_dict["experimental_2024"].update(hyperparameter_config_dict["zeroshot_2023"])
 hyperparameter_config_dict["experimental"] = hyperparameter_config_dict["experimental_2024"]
 
+
 def get_hyperparameter_config_options():
     return list(hyperparameter_config_dict.keys())
 
@@ -140,5 +175,7 @@ def get_hyperparameter_config_options():
 def get_hyperparameter_config(config_name):
     config_options = get_hyperparameter_config_options()
     if config_name not in config_options:
-        raise ValueError(f"Valid hyperparameter config names are: {config_options}, but '{config_name}' was given instead.")
+        raise ValueError(
+            f"Valid hyperparameter config names are: {config_options}, but '{config_name}' was given instead."
+        )
     return copy.deepcopy(hyperparameter_config_dict[config_name])

{autogluon_tabular-1.4.1b20251201 → autogluon_tabular-1.5.1b20260114}/src/autogluon/tabular/configs/presets_configs.py

@@ -9,6 +9,19 @@ tabular_presets_dict = dict(
         "hyperparameters": "zeroshot",
         "time_limit": 3600,
     },
+    best_quality_v150={
+        "auto_stack": True,
+        "dynamic_stacking": "auto",
+        "num_stack_levels": 0,
+        "hyperparameters": "zeroshot_2025_12_18_cpu",
+        "time_limit": 3600,
+        "callbacks": [
+            [
+                "EarlyStoppingCountCallback",
+                {"patience": [[100, 4], [500, 8], [2500, 15], [10000, 40], [100000, 100], None]},
+            ]
+        ],
+    },
     # High predictive accuracy with fast inference. ~8x faster inference and ~8x lower disk usage than `best_quality`.
     # Recommended for applications that require fast inference speed and/or small model size.
     # Aliases: high
@@ -21,6 +34,22 @@ tabular_presets_dict = dict(
         "set_best_to_refit_full": True,
         "save_bag_folds": False,
     },
+    high_quality_v150={
+        "auto_stack": True,
+        "dynamic_stacking": "auto",
+        "num_stack_levels": 0,
+        "hyperparameters": "zeroshot_2025_12_18_cpu",
+        "time_limit": 3600,
+        "callbacks": [
+            [
+                "EarlyStoppingCountCallback",
+                {"patience": [[100, 4], [500, 8], [2500, 15], [10000, 40], [100000, 100], None]},
+            ]
+        ],
+        "refit_full": True,
+        "set_best_to_refit_full": True,
+        "save_bag_folds": False,
+    },
     # Good predictive accuracy with very fast inference. ~4x faster training, ~8x faster inference and ~8x lower disk usage than `high_quality`.
     # Recommended for applications that require very fast inference speed.
     # Aliases: good
@@ -46,7 +75,13 @@ tabular_presets_dict = dict(
     optimize_for_deployment={"keep_only_best": True, "save_space": True},
     # Disables automated feature generation when text features are detected.
     # This is useful to determine how beneficial text features are to the end result, as well as to ensure features are not mistaken for text when they are not.
-    ignore_text={"_feature_generator_kwargs": {"enable_text_ngram_features": False, "enable_text_special_features": False, "enable_raw_text_features": False}},
+    ignore_text={
+        "_feature_generator_kwargs": {
+            "enable_text_ngram_features": False,
+            "enable_text_special_features": False,
+            "enable_raw_text_features": False,
+        }
+    },
     ignore_text_ngrams={"_feature_generator_kwargs": {"enable_text_ngram_features": False}},
     # Fit only interpretable models.
     interpretable={
@@ -64,25 +99,48 @@ tabular_presets_dict = dict(
     best_quality_v082={"auto_stack": True},
     # High predictive accuracy with fast inference. ~10x-200x faster inference and ~10x-200x lower disk usage than `best_quality`.
     # Recommended for applications that require reasonable inference speed and/or model size.
-    high_quality_v082={"auto_stack": True, "refit_full": True, "set_best_to_refit_full": True, "save_bag_folds": False},
+    high_quality_v082={
+        "auto_stack": True,
+        "refit_full": True,
+        "set_best_to_refit_full": True,
+        "save_bag_folds": False,
+    },
     # Good predictive accuracy with very fast inference. ~4x faster inference and ~4x lower disk usage than `high_quality`.
     # Recommended for applications that require fast inference speed.
-    good_quality_v082={"auto_stack": True, "refit_full": True, "set_best_to_refit_full": True, "save_bag_folds": False, "hyperparameters": "light"},
+    good_quality_v082={
+        "auto_stack": True,
+        "refit_full": True,
+        "set_best_to_refit_full": True,
+        "save_bag_folds": False,
+        "hyperparameters": "light",
+    },
     # ------------------------------------------
     # Experimental presets. Only use these presets if you are ok with unstable and potentially poor performing presets.
     #  Experimental presets can be removed or changed without warning.
-
     # [EXPERIMENTAL PRESET] The `extreme` preset may be changed or removed without warning.
     # This preset acts as a testing ground for cutting edge features and models which could later be added to the `best_quality` preset in future releases.
     # Using this preset can lead to unexpected crashes, as it hasn't been as thoroughly tested as other presets.
     # Absolute best predictive accuracy with **zero** consideration to inference time or disk usage.
     # Recommended for applications that benefit from the best possible model accuracy and **do not** care about inference speed.
     # Significantly stronger than `best_quality`, but can be over 10x slower in inference.
-    # Uses pre-trained tabular foundation models, which add a minimum of 1-2 GB to the predictor artifact's size.
+    # Uses pre-trained tabular foundation models, which add a minimum of 100 MB to the predictor artifact's size.
     # For best results, use as large of an instance as possible with a GPU and as many CPU cores as possible (ideally 64+ cores)
     # Aliases: extreme, experimental, experimental_quality
     # GPU STRONGLY RECOMMENDED
     extreme_quality={
+        "auto_stack": True,
+        "dynamic_stacking": "auto",
+        "num_stack_levels": 0,
+        "hyperparameters": "zeroshot_2025_12_18_gpu",
+        "time_limit": 3600,
+        "callbacks": [
+            [
+                "EarlyStoppingCountCallback",
+                {"patience": [[100, 4], [500, 8], [2500, 15], [10000, 40], [100000, 100], None]},
+            ]
+        ],
+    },
+    extreme_quality_v140={
         "auto_stack": True,
         "dynamic_stacking": "auto",
         "num_bag_sets": 1,
@@ -90,7 +148,6 @@ tabular_presets_dict = dict(
         "hyperparameters": None,
         "time_limit": 3600,
     },
-
     # Preset with a portfolio learned from TabArena v0.1: https://tabarena.ai/
     # Uses tabular foundation models: TabPFNv2, TabICL, Mitra
     # Uses deep learning model: TabM
@@ -105,7 +162,6 @@ tabular_presets_dict = dict(
         "hyperparameters": "zeroshot_2025_tabfm",
         "time_limit": 3600,
     },
-
     # DOES NOT SUPPORT GPU.
     experimental_quality_v120={
         "auto_stack": True,
@@ -116,7 +172,6 @@ tabular_presets_dict = dict(
         "num_gpus": 0,
         "time_limit": 3600,
     },
-
     # ------------------------------------------
     # ------------------------------------------
     # ------------------------------------------
@@ -140,5 +195,11 @@ tabular_presets_alias = dict(
     mq="medium_quality",
     experimental="extreme_quality",
     experimental_quality="extreme_quality",
-    experimental_quality_v140="extreme_quality",
+    experimental_quality_v140="extreme_quality_v140",
+    best_v140="best_quality",
+    best_v150="best_quality_v150",
+    best_quality_v140="best_quality",
+    high_v150="high_quality_v150",
+    extreme_v140="extreme_quality_v140",
+    extreme_v150="extreme_quality",
 )