autogluon.tabular 1.4.0__py3-none-any.whl → 1.4.1b20251128__py3-none-any.whl

autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py

@@ -50,6 +50,7 @@ class TabularNeuralNetTorchModel(AbstractNeuralNetworkModel):
     ag_key = "NN_TORCH"
     ag_name = "NeuralNetTorch"
     ag_priority = 25
+    seed_name = "seed_value"
 
     # Constants used throughout this class:
     unique_category_str = np.nan  # string used to represent missing values and unknown categories for categorical features.
@@ -191,7 +192,7 @@ class TabularNeuralNetTorchModel(AbstractNeuralNetworkModel):
 
         processor_kwargs, optimizer_kwargs, fit_kwargs, loss_kwargs, params = self._prepare_params(params=params)
 
-        seed_value = params.pop("seed_value", 0)
+        seed_value = params.pop(self.seed_name, self.default_random_seed)
 
         self._num_cpus_infer = params.pop("_num_cpus_infer", 1)
         if seed_value is not None:  # Set seeds

autogluon/tabular/models/xgboost/xgboost_model.py

@@ -32,6 +32,7 @@ class XGBoostModel(AbstractModel):
     ag_key = "XGB"
     ag_name = "XGBoost"
     ag_priority = 40
+    seed_name = "seed"
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
@@ -182,12 +183,18 @@ class XGBoostModel(AbstractModel):
         from xgboost import XGBClassifier, XGBRegressor
 
         model_type = XGBClassifier if self.problem_type in PROBLEM_TYPES_CLASSIFICATION else XGBRegressor
-        self.model = model_type(**params)
+
         import warnings
 
         with warnings.catch_warnings():
             # FIXME: v1.1: Upgrade XGBoost to 2.0.1+ to avoid deprecation warnings from Pandas 2.1+ during XGBoost fit.
             warnings.simplefilter(action="ignore", category=FutureWarning)
+            if params.get("device", "cpu") == "cuda:0":
+                # verbosity=0 to hide UserWarning: Falling back to prediction using DMatrix due to mismatched devices.
+                # TODO: Find a way to hide this warning without setting verbosity=0
+                #  ref: https://github.com/dmlc/xgboost/issues/9791
+                params["verbosity"] = 0
+            self.model = model_type(**params)
             self.model.fit(X=X, y=y, eval_set=eval_set, verbose=False, sample_weight=sample_weight)
 
         if generate_curves:

autogluon/tabular/predictor/predictor.py

@@ -20,6 +20,7 @@ from autogluon.common import FeatureMetadata, TabularDataset
 from autogluon.common.loaders import load_json
 from autogluon.common.savers import save_json
 from autogluon.common.utils.file_utils import get_directory_size, get_directory_size_per_file
+from autogluon.common.utils.resource_utils import ResourceManager, get_resource_manager
 from autogluon.common.utils.hyperparameter_utils import get_hyperparameter_str_deprecation_msg, is_advanced_hyperparameter_format
 from autogluon.common.utils.log_utils import add_log_to_file, set_logger_verbosity, warn_if_mlflow_autologging_is_enabled
 from autogluon.common.utils.pandas_utils import get_approximate_df_mem_usage
@@ -48,10 +49,14 @@ from autogluon.core.utils import get_pred_from_proba_df, plot_performance_vs_tri
 from autogluon.core.utils.decorators import apply_presets
 from autogluon.core.utils.loaders import load_pkl, load_str
 from autogluon.core.utils.savers import save_pkl, save_str
-from autogluon.core.utils.utils import CVSplitter, default_holdout_frac, generate_train_test_split_combined
+from autogluon.core.utils.utils import CVSplitter, generate_train_test_split_combined
 
 from ..configs.feature_generator_presets import get_default_feature_generator
 from ..configs.hyperparameter_configs import get_hyperparameter_config
+from ..configs.pipeline_presets import (
+    USE_BAG_HOLDOUT_AUTO_THRESHOLD,
+    get_validation_and_stacking_method,
+)
 from ..configs.presets_configs import tabular_presets_alias, tabular_presets_dict
 from ..learner import AbstractTabularLearner, DefaultLearner
 from ..trainer.abstract_trainer import AbstractTabularTrainer
@@ -165,6 +170,10 @@ class TabularPredictor:
             trainer_type : AbstractTabularTrainer, default = AutoTrainer
                 A class inheriting from `AbstractTabularTrainer` that controls training/ensembling of many models.
                 If you don't know what this is, keep it as the default.
+        default_base_path : str | Path | None, default = None
+            A default base path to use for the time-stamped folder if `path` is None.
+            If None, defaults to `AutogluonModels`. Only used if `path` is None, and thus
+            only used for local paths, not s3 paths.
     """
 
     Dataset = TabularDataset
@@ -201,7 +210,7 @@ class TabularPredictor:
                 f"We do not recommend specifying weight_evaluation when sample_weight='{self.sample_weight}', instead specify appropriate eval_metric."
             )
         self._validate_init_kwargs(kwargs)
-        path = setup_outputdir(path)
+        path = setup_outputdir(path=path, default_base_path=kwargs.get("default_base_path"))
 
         learner_type = kwargs.get("learner_type", DefaultLearner)
         learner_kwargs = kwargs.get("learner_kwargs", dict())
@@ -425,12 +434,17 @@ class TabularPredictor:
             Table of the training data as a pandas DataFrame.
             If str is passed, `train_data` will be loaded using the str value as the file path.
         tuning_data : :class:`pd.DataFrame` or str, optional
-            Another dataset containing validation data reserved for tuning processes such as early stopping and hyperparameter tuning.
+            Another dataset containing validation data reserved for tuning processes such as early stopping, hyperparameter tuning, and ensembling.
             This dataset should be in the same format as `train_data`.
             If str is passed, `tuning_data` will be loaded using the str value as the file path.
-            Note: final model returned may be fit on `tuning_data` as well as `train_data`. Do not provide your evaluation test data here!
-            In particular, when `num_bag_folds` > 0 or `num_stack_levels` > 0, models will be trained on both `tuning_data` and `train_data`.
-            If `tuning_data = None`, `fit()` will automatically hold out some random validation examples from `train_data`.
+            Note: If `refit_full=True` is specified, the final model may be fit on `tuning_data` as well as `train_data`.
+            Note: Because `tuning_data` is used to determine which model is the 'best' model, as well as to determine the ensemble weights,
+                it should not be considered a fully unseen dataset. It is possible that AutoGluon will be overfit to the `tuning_data`.
+                To ensure an unbiased evaluation, use separate unseen test data to evaluate the final model using `predictor.leaderboard(test_data, display=True)`.
+                Do not provide your evaluation test data as `tuning_data`!
+            If bagging is not enabled and `tuning_data = None`: `fit()` will automatically hold out some random validation samples from `train_data`.
+            If bagging is enabled  and `tuning_data = None`: no tuning data will be used. Instead, AutoGluon will perform cross-validation.
+            If bagging is enabled: `use_bag_holdout=True` must be specified in order to provide tuning data. If specified, AutoGluon will still perform cross-validation for model fits, but will use `tuning_data` for optimizing the weighted ensemble weights and model calibration.
         time_limit : int, default = None
             Approximately how long `fit()` should run for (wallclock time in seconds).
             If not specified, `fit()` will run until all models have completed training, but will not repeatedly bag models unless `num_bag_sets` is specified.
@@ -512,6 +526,7 @@ class TabularPredictor:
                     'GBM' (LightGBM)
                     'CAT' (CatBoost)
                     'XGB' (XGBoost)
+                    'EBM' (Explainable Boosting Machine)
                     'REALMLP' (RealMLP)
                     'TABM' (TabM)
                     'MITRA' (Mitra)
@@ -1077,7 +1092,8 @@ class TabularPredictor:
             elif verbosity >= 4:
                 logger.log(20, f"Verbosity: {verbosity} (Maximum Logging)")
 
-        include_gpu_count = verbosity >= 3
+        resource_manager: ResourceManager = get_resource_manager()
+        include_gpu_count = resource_manager.get_gpu_count_torch() or verbosity >= 3
         sys_msg = get_ag_system_info(path=self.path, include_gpu_count=include_gpu_count)
         logger.log(20, sys_msg)
 
@@ -1117,10 +1133,6 @@ class TabularPredictor:
         self._validate_calibrate_decision_threshold(calibrate_decision_threshold=calibrate_decision_threshold)
         self._validate_fit_strategy(fit_strategy=fit_strategy)
 
-        holdout_frac = kwargs["holdout_frac"]
-        num_bag_folds = kwargs["num_bag_folds"]
-        num_bag_sets = kwargs["num_bag_sets"]
-        num_stack_levels = kwargs["num_stack_levels"]
         auto_stack = kwargs["auto_stack"]
         feature_generator = kwargs["feature_generator"]
         unlabeled_data = kwargs["unlabeled_data"]
@@ -1216,16 +1228,46 @@ class TabularPredictor:
         else:
             ag_args_fit = learning_curves
 
+        use_bag_holdout_was_auto = False
+        dynamic_stacking_was_auto = False
+        if isinstance(use_bag_holdout,str) and use_bag_holdout == "auto":
+            use_bag_holdout = None
+            use_bag_holdout_was_auto = True
+        if isinstance(dynamic_stacking,str) and dynamic_stacking == "auto":
+            dynamic_stacking = None
+            dynamic_stacking_was_auto = True
+
+        (
+            num_bag_folds,
+            num_bag_sets,
+            num_stack_levels,
+            dynamic_stacking,
+            use_bag_holdout,
+            holdout_frac,
+            refit_full,
+        ) = get_validation_and_stacking_method(
+            num_bag_folds=kwargs["num_bag_folds"],
+            num_bag_sets=kwargs["num_bag_sets"],
+            use_bag_holdout=use_bag_holdout,
+            holdout_frac=kwargs["holdout_frac"],
+            auto_stack=auto_stack,
+            num_stack_levels=kwargs["num_stack_levels"],
+            dynamic_stacking=dynamic_stacking,
+            refit_full=kwargs["refit_full"],
+            num_train_rows=len(train_data),
+            problem_type=inferred_problem_type,
+            hpo_enabled=ag_args.get("hyperparameter_tune_kwargs", None) is not None,
+        )
+
         num_bag_folds, num_bag_sets, num_stack_levels, dynamic_stacking, use_bag_holdout = self._sanitize_stack_args(
             num_bag_folds=num_bag_folds,
             num_bag_sets=num_bag_sets,
             num_stack_levels=num_stack_levels,
-            time_limit=time_limit,
-            auto_stack=auto_stack,
             num_train_rows=len(train_data),
-            problem_type=inferred_problem_type,
             dynamic_stacking=dynamic_stacking,
             use_bag_holdout=use_bag_holdout,
+            use_bag_holdout_was_auto=use_bag_holdout_was_auto,
+            dynamic_stacking_was_auto=dynamic_stacking_was_auto,
         )
         if auto_stack:
             logger.log(
@@ -1234,9 +1276,6 @@ class TabularPredictor:
                 f"num_stack_levels={num_stack_levels}, num_bag_folds={num_bag_folds}, num_bag_sets={num_bag_sets}",
             )
 
-        if holdout_frac is None:
-            holdout_frac = default_holdout_frac(len(train_data), ag_args.get("hyperparameter_tune_kwargs", None) is not None)
-
         if kwargs["save_bag_folds"] is not None and kwargs["_save_bag_folds"] is not None:
             raise ValueError(
                 f"Cannot specify both `save_bag_folds` and `_save_bag_folds` at the same time. "
@@ -1324,7 +1363,7 @@ class TabularPredictor:
         )
         ag_post_fit_kwargs = dict(
             keep_only_best=kwargs["keep_only_best"],
-            refit_full=kwargs["refit_full"],
+            refit_full=refit_full,
             set_best_to_refit_full=kwargs["set_best_to_refit_full"],
             save_space=kwargs["save_space"],
             calibrate=kwargs["calibrate"],
@@ -1593,7 +1632,6 @@ class TabularPredictor:
             if _ds_ray is not None:
                 # Handle resources
                 # FIXME: what about distributed?
-                from autogluon.common.utils.resource_utils import ResourceManager
 
                 total_resources = ag_fit_kwargs["core_kwargs"]["total_resources"]
 
@@ -5061,6 +5099,7 @@ class TabularPredictor:
             "learner_type",
             "learner_kwargs",
             "quantile_levels",
+            "default_base_path",
         }
         invalid_keys = []
         for key in kwargs:
@@ -5484,41 +5523,12 @@ class TabularPredictor:
         num_bag_folds: int,
         num_bag_sets: int,
         num_stack_levels: int,
-        time_limit: float | None,
-        auto_stack: bool,
         num_train_rows: int,
-        problem_type: str,
         dynamic_stacking: bool | str,
         use_bag_holdout: bool | str,
+        use_bag_holdout_was_auto: bool,
+        dynamic_stacking_was_auto: bool,
     ):
-        use_bag_holdout_auto_threshold = 1000000
-        use_bag_holdout_was_auto = False
-        dynamic_stacking_was_auto = False
-        if isinstance(use_bag_holdout, str) and use_bag_holdout == "auto":
-            # Leverage use_bag_holdout when data is large to safeguard against stack leakage
-            use_bag_holdout = num_train_rows >= use_bag_holdout_auto_threshold
-            use_bag_holdout_was_auto = True
-        if isinstance(dynamic_stacking, str) and dynamic_stacking == "auto":
-            dynamic_stacking = not use_bag_holdout
-            dynamic_stacking_was_auto = True
-        if auto_stack:
-            # TODO: What about datasets that are 100k+? At a certain point should we not bag?
-            # TODO: What about time_limit? Metalearning can tell us expected runtime of each model, then we can select optimal folds + stack levels to fit time constraint
-            if num_bag_folds is None:
-                num_bag_folds = min(8, max(5, math.floor(num_train_rows / 10)))
-            if num_stack_levels is None:
-                if dynamic_stacking:
-                    num_stack_levels = 1
-                else:
-                    if use_bag_holdout or problem_type != BINARY:
-                        num_stack_levels = min(1, max(0, math.floor(num_train_rows / 750)))
-                    else:
-                        # Disable multi-layer stacking to avoid stack info leakage
-                        num_stack_levels = 0
-        if num_bag_folds is None:
-            num_bag_folds = 0
-        if num_stack_levels is None:
-            num_stack_levels = 0
         if not isinstance(num_bag_folds, int):
             raise ValueError(f"num_bag_folds must be an integer. (num_bag_folds={num_bag_folds})")
         if not isinstance(num_stack_levels, int):
@@ -5527,8 +5537,6 @@ class TabularPredictor:
             raise ValueError(f"num_bag_folds must be equal to 0 or >=2. (num_bag_folds={num_bag_folds})")
         if num_stack_levels != 0 and num_bag_folds == 0:
             raise ValueError(f"num_stack_levels must be 0 if num_bag_folds is 0. (num_stack_levels={num_stack_levels}, num_bag_folds={num_bag_folds})")
-        if num_bag_sets is None:
-            num_bag_sets = 1
         if not isinstance(num_bag_sets, int):
             raise ValueError(f"num_bag_sets must be an integer. (num_bag_sets={num_bag_sets})")
         if not isinstance(dynamic_stacking, bool):
@@ -5538,11 +5546,11 @@ class TabularPredictor:
 
         if use_bag_holdout_was_auto and num_bag_folds != 0:
             if use_bag_holdout:
-                log_extra = f"Reason: num_train_rows >= {use_bag_holdout_auto_threshold}. (num_train_rows={num_train_rows})"
+                log_extra = f"Reason: num_train_rows >= {USE_BAG_HOLDOUT_AUTO_THRESHOLD}. (num_train_rows={num_train_rows})"
             else:
-                log_extra = f"Reason: num_train_rows < {use_bag_holdout_auto_threshold}. (num_train_rows={num_train_rows})"
+                log_extra = f"Reason: num_train_rows < {USE_BAG_HOLDOUT_AUTO_THRESHOLD}. (num_train_rows={num_train_rows})"
             logger.log(20, f"Setting use_bag_holdout from 'auto' to {use_bag_holdout}. {log_extra}")
-        log_extra_ds = None
+
         if dynamic_stacking and num_stack_levels < 1:
             log_extra_ds = f"Reason: Stacking is not enabled. (num_stack_levels={num_stack_levels})"
             if not dynamic_stacking_was_auto:

autogluon/tabular/registry/_ag_model_registry.py

@@ -8,6 +8,7 @@ from . import ModelRegistry
 from ..models import (
     BoostedRulesModel,
     CatBoostModel,
+    EBMModel,
     FastTextModel,
     FigsModel,
     FTTransformerModel,
@@ -64,6 +65,7 @@ REGISTERED_MODEL_CLS_LST = [
     HSTreeModel,
     BoostedRulesModel,
     DummyModel,
+    EBMModel,
 ]
 
 # TODO: Replace logic in `autogluon.tabular.trainer.model_presets.presets` with `ag_model_registry`

autogluon/tabular/testing/fit_helper.py

@@ -175,6 +175,7 @@ class FitHelper:
         use_test_for_val: bool = False,
         raise_on_model_failure: bool | None = None,
         deepcopy_fit_args: bool = True,
+        verify_model_seed: bool = False,
     ) -> TabularPredictor:
         if compiler_configs is None:
             compiler_configs = {}
@@ -269,6 +270,11 @@ class FitHelper:
                 assert not model_info["val_in_fit"], f"val data must not be present in refit model if `can_refit_full=True`. Maybe an exception occurred?"
             else:
                 assert model_info["val_in_fit"], f"val data must be present in refit model if `can_refit_full=False`"
+        if verify_model_seed:
+            model_names = predictor.model_names()
+            for model_name in model_names:
+                model = predictor._trainer.load_model(model_name)
+                _verify_model_seed(model=model)
 
         if predictor_info:
             predictor.info()
@@ -339,6 +345,7 @@ class FitHelper:
         require_known_problem_types: bool = True,
         raise_on_model_failure: bool = True,
         problem_types: list[str] | None = None,
+        verify_model_seed: bool = True,
         **kwargs,
     ):
         """
@@ -355,12 +362,18 @@ class FitHelper:
         problem_types: list[str], optional
             If specified, checks the given problem_types.
             If None, checks `model_cls.supported_problem_types()`
+        verify_model_seed: bool = True
         **kwargs
 
         Returns
         -------
 
         """
+        if verify_model_seed and model_cls.seed_name is not None:
+            # verify that the seed logic works
+            model_hyperparameters = model_hyperparameters.copy()
+            model_hyperparameters[model_cls.seed_name] = 42
+
         fit_args = dict(
             hyperparameters={model_cls: model_hyperparameters},
         )
@@ -429,6 +442,7 @@ class FitHelper:
                     refit_full=refit_full,
                     extra_metrics=_extra_metrics,
                     raise_on_model_failure=raise_on_model_failure,
+                    verify_model_seed=verify_model_seed,
                     **kwargs,
                 )
 
@@ -460,6 +474,7 @@ class FitHelper:
                         refit_full=refit_full,
                         extra_metrics=_extra_metrics,
                         raise_on_model_failure=raise_on_model_failure,
+                        verify_model_seed=verify_model_seed,
                         **kwargs,
                     )
 
@@ -476,3 +491,16 @@ def stacked_overfitting_assert(
     if expected_stacked_overfitting_at_test is not None:
         stacked_overfitting = check_stacked_overfitting_from_leaderboard(lb)
         assert stacked_overfitting == expected_stacked_overfitting_at_test, "Expected stacked overfitting at test mismatch!"
+
+
+def _verify_model_seed(model: AbstractModel):
+    assert model.random_seed is None or isinstance(model.random_seed, int)
+    if model.seed_name is not None:
+        if model.seed_name in model._user_params:
+            assert model.random_seed == model._user_params[model.seed_name]
+        assert model.seed_name in model.params
+        assert model.random_seed == model.params[model.seed_name]
+    if isinstance(model, BaggedEnsembleModel):
+        for child in model.models:
+            child = model.load_child(child)
+            _verify_model_seed(child)

autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.4.0"
+__version__ = "1.4.1b20251128"
 __lite__ = False

autogluon.tabular-1.4.1b20251128-py3.11-nspkg.pth

@@ -0,0 +1 @@
+import sys, types, os;p = os.path.join(sys._getframe(1).f_locals['sitedir'], *('autogluon',));importlib = __import__('importlib.util');__import__('importlib.machinery');m = sys.modules.setdefault('autogluon', importlib.util.module_from_spec(importlib.machinery.PathFinder.find_spec('autogluon', [os.path.dirname(p)])));m = m or sys.modules.setdefault('autogluon', types.ModuleType('autogluon'));mp = (m or []) and m.__dict__.setdefault('__path__',[]);(p not in mp) and mp.append(p)

{autogluon.tabular-1.4.0.dist-info → autogluon_tabular-1.4.1b20251128.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: autogluon.tabular
-Version: 1.4.0
+Version: 1.4.1b20251128
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -9,8 +9,7 @@ Project-URL: Documentation, https://auto.gluon.ai
 Project-URL: Bug Reports, https://github.com/autogluon/autogluon/issues
 Project-URL: Source, https://github.com/autogluon/autogluon/
 Project-URL: Contribute!, https://github.com/autogluon/autogluon/blob/master/CONTRIBUTING.md
-Platform: UNKNOWN
-Classifier: Development Status :: 5 - Production/Stable
+Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Education
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -24,114 +23,130 @@ Classifier: Operating System :: Microsoft :: Windows
 Classifier: Operating System :: POSIX
 Classifier: Operating System :: Unix
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Classifier: Topic :: Scientific/Engineering :: Image Recognition
-Requires-Python: >=3.9, <3.13
+Requires-Python: >=3.10, <3.14
 Description-Content-Type: text/markdown
-License-File: ../LICENSE
-License-File: ../NOTICE
+License-File: LICENSE
+License-File: NOTICE
 Requires-Dist: numpy<2.4.0,>=1.25.0
 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.0
-Requires-Dist: autogluon.features==1.4.0
-Provides-Extra: all
-Requires-Dist: catboost<1.3,>=1.2; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
-Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
-Requires-Dist: loguru; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
-Requires-Dist: einx; extra == "all"
-Requires-Dist: autogluon.core[all]==1.4.0; extra == "all"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
-Requires-Dist: transformers; extra == "all"
-Requires-Dist: spacy<3.9; extra == "all"
-Requires-Dist: torch<2.8,>=2.2; extra == "all"
-Requires-Dist: omegaconf; extra == "all"
-Requires-Dist: huggingface-hub[torch]; extra == "all"
-Requires-Dist: blis<1.2.1,>=0.7.0; (platform_system == "Windows" and python_version == "3.9") and extra == "all"
+Requires-Dist: autogluon.core==1.4.1b20251128
+Requires-Dist: autogluon.features==1.4.1b20251128
+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"
+Provides-Extra: realmlp
+Requires-Dist: pytabkit<1.7,>=1.6; 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.2; extra == "fastai"
+Requires-Dist: torch<2.8,>=2.6; extra == "fastai"
 Requires-Dist: fastai<2.9,>=2.3.1; extra == "fastai"
-Requires-Dist: blis<1.2.1,>=0.7.0; (platform_system == "Windows" and python_version == "3.9") and extra == "fastai"
-Provides-Extra: imodels
-Requires-Dist: imodels<2.1.0,>=1.3.10; extra == "imodels"
-Provides-Extra: lightgbm
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "lightgbm"
+Provides-Extra: tabm
+Requires-Dist: torch<2.8,>=2.6; extra == "tabm"
+Provides-Extra: tabpfn
+Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabpfn"
+Provides-Extra: tabpfnmix
+Requires-Dist: torch<2.8,>=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.2; extra == "mitra"
+Requires-Dist: torch<2.8,>=2.6; extra == "mitra"
 Requires-Dist: transformers; extra == "mitra"
-Requires-Dist: huggingface-hub[torch]; 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"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.4.0; extra == "ray"
-Provides-Extra: realmlp
-Requires-Dist: pytabkit<1.7,>=1.6; extra == "realmlp"
+Requires-Dist: autogluon.core[all]==1.4.1b20251128; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.5,>=2024.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: onnx<1.18.0,>=1.13.0; platform_system != "Windows" and extra == "skl2onnx"
-Requires-Dist: onnx<1.16.2,>=1.13.0; platform_system == "Windows" and extra == "skl2onnx"
+Provides-Extra: all
+Requires-Dist: omegaconf; extra == "all"
+Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
+Requires-Dist: spacy<3.9; extra == "all"
+Requires-Dist: einops<0.9,>=0.7; extra == "all"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: einx; extra == "all"
+Requires-Dist: autogluon.core[all]==1.4.1b20251128; extra == "all"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
+Requires-Dist: torch<2.8,>=2.6; extra == "all"
+Requires-Dist: loguru; extra == "all"
+Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: transformers; extra == "all"
+Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
+Requires-Dist: huggingface_hub[torch]<1.0; extra == "all"
 Provides-Extra: tabarena
 Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
+Requires-Dist: omegaconf; extra == "tabarena"
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "tabarena"
-Requires-Dist: loguru; extra == "tabarena"
+Requires-Dist: spacy<3.9; extra == "tabarena"
+Requires-Dist: einops<0.9,>=0.7; extra == "tabarena"
+Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabarena"
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
 Requires-Dist: einx; extra == "tabarena"
-Requires-Dist: autogluon.core[all]==1.4.0; extra == "tabarena"
+Requires-Dist: autogluon.core[all]==1.4.1b20251128; extra == "tabarena"
+Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "tabarena"
+Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
+Requires-Dist: loguru; extra == "tabarena"
+Requires-Dist: pytabkit<1.7,>=1.6; extra == "tabarena"
+Requires-Dist: torch<2.8,>=2.6; extra == "tabarena"
+Requires-Dist: transformers; extra == "tabarena"
 Requires-Dist: tabicl<0.2,>=0.1.3; extra == "tabarena"
 Requires-Dist: xgboost<3.1,>=2.0; extra == "tabarena"
-Requires-Dist: transformers; extra == "tabarena"
-Requires-Dist: spacy<3.9; extra == "tabarena"
-Requires-Dist: torch<2.8,>=2.2; extra == "tabarena"
-Requires-Dist: omegaconf; extra == "tabarena"
-Requires-Dist: huggingface-hub[torch]; extra == "tabarena"
-Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabarena"
-Requires-Dist: pytabkit<1.7,>=1.6; extra == "tabarena"
-Requires-Dist: blis<1.2.1,>=0.7.0; (platform_system == "Windows" and python_version == "3.9") and extra == "tabarena"
-Provides-Extra: tabicl
-Requires-Dist: tabicl<0.2,>=0.1.3; extra == "tabicl"
-Provides-Extra: tabm
-Requires-Dist: torch<2.8,>=2.2; extra == "tabm"
-Provides-Extra: tabpfn
-Requires-Dist: tabpfn<2.2,>=2.0.9; extra == "tabpfn"
-Provides-Extra: tabpfnmix
-Requires-Dist: torch<2.8,>=2.2; extra == "tabpfnmix"
-Requires-Dist: huggingface-hub[torch]; extra == "tabpfnmix"
-Requires-Dist: einops<0.9,>=0.7; extra == "tabpfnmix"
+Requires-Dist: huggingface_hub[torch]<1.0; 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.2; extra == "tests"
-Requires-Dist: huggingface-hub[torch]; extra == "tests"
+Requires-Dist: torch<2.8,>=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: onnx<1.18.0,>=1.13.0; platform_system != "Windows" and extra == "tests"
-Requires-Dist: onnx<1.16.2,>=1.13.0; platform_system == "Windows" and extra == "tests"
-Provides-Extra: xgboost
-Requires-Dist: xgboost<3.1,>=2.0; extra == "xgboost"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 
 
@@ -142,7 +157,7 @@ Requires-Dist: xgboost<3.1,>=2.0; extra == "xgboost"
 
 [![Latest Release](https://img.shields.io/github/v/release/autogluon/autogluon)](https://github.com/autogluon/autogluon/releases)
 [![Conda Forge](https://img.shields.io/conda/vn/conda-forge/autogluon.svg)](https://anaconda.org/conda-forge/autogluon)
-[![Python Versions](https://img.shields.io/badge/python-3.9%20%7C%203.10%20%7C%203.11%20%7C%203.12-blue)](https://pypi.org/project/autogluon/)
+[![Python Versions](https://img.shields.io/badge/python-3.10%20%7C%203.11%20%7C%203.12%20%7C%203.13-blue)](https://pypi.org/project/autogluon/)
 [![Downloads](https://pepy.tech/badge/autogluon/month)](https://pepy.tech/project/autogluon)
 [![GitHub license](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](./LICENSE)
 [![Discord](https://img.shields.io/discord/1043248669505368144?color=7289da&label=Discord&logo=discord&logoColor=ffffff)](https://discord.gg/wjUmjqAc2N)
@@ -159,7 +174,7 @@ AutoGluon, developed by AWS AI, automates machine learning tasks enabling you to
 
 ## 💾 Installation
 
-AutoGluon is supported on Python 3.9 - 3.12 and is available on Linux, MacOS, and Windows.
+AutoGluon is supported on Python 3.10 - 3.13 and is available on Linux, MacOS, and Windows.
 
 You can install AutoGluon with:
 
@@ -182,8 +197,8 @@ predictions = predictor.predict("test.csv")
 | AutoGluon Task      |                                                                                Quickstart                                                                                |                                                                                API                                                                                |
 |:--------------------|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|:-----------------------------------------------------------------------------------------------------------------------------------------------------------------:|
 | TabularPredictor    | [![Quick Start](https://img.shields.io/static/v1?label=&message=tutorial&color=grey)](https://auto.gluon.ai/stable/tutorials/tabular/tabular-quick-start.html) |                 [![API](https://img.shields.io/badge/api-reference-blue.svg)](https://auto.gluon.ai/stable/api/autogluon.tabular.TabularPredictor.html)                 |
-| MultiModalPredictor | [![Quick Start](https://img.shields.io/static/v1?label=&message=tutorial&color=grey)](https://auto.gluon.ai/stable/tutorials/multimodal/multimodal_prediction/multimodal-quick-start.html)            | [![API](https://img.shields.io/badge/api-reference-blue.svg)](https://auto.gluon.ai/stable/api/autogluon.multimodal.MultiModalPredictor.html) |
 | TimeSeriesPredictor | [![Quick Start](https://img.shields.io/static/v1?label=&message=tutorial&color=grey)](https://auto.gluon.ai/stable/tutorials/timeseries/forecasting-quick-start.html)            | [![API](https://img.shields.io/badge/api-reference-blue.svg)](https://auto.gluon.ai/stable/api/autogluon.timeseries.TimeSeriesPredictor.html) |
+| MultiModalPredictor | [![Quick Start](https://img.shields.io/static/v1?label=&message=tutorial&color=grey)](https://auto.gluon.ai/stable/tutorials/multimodal/multimodal_prediction/multimodal-quick-start.html)            | [![API](https://img.shields.io/badge/api-reference-blue.svg)](https://auto.gluon.ai/stable/api/autogluon.multimodal.MultiModalPredictor.html) |
 
 ## :mag: Resources
 
@@ -206,7 +221,10 @@ Below is a curated list of recent tutorials and talks on AutoGluon. A comprehens
 - [Benchmarking Multimodal AutoML for Tabular Data with Text Fields](https://datasets-benchmarks-proceedings.neurips.cc/paper/2021/file/9bf31c7ff062936a96d3c8bd1f8f2ff3-Paper-round2.pdf) (*NeurIPS*, 2021) ([BibTeX](CITING.md#autogluonmultimodal))
 - [XTab: Cross-table Pretraining for Tabular Transformers](https://proceedings.mlr.press/v202/zhu23k/zhu23k.pdf) (*ICML*, 2023)
 - [AutoGluon-TimeSeries: AutoML for Probabilistic Time Series Forecasting](https://arxiv.org/abs/2308.05566) (*AutoML Conf*, 2023) ([BibTeX](CITING.md#autogluontimeseries))
-- [TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applications](https://arxiv.org/pdf/2311.02971.pdf) (*Under Review*, 2024)
+- [TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applications](https://arxiv.org/pdf/2311.02971.pdf) (*AutoML Conf*, 2024)
+- [AutoGluon-Multimodal (AutoMM): Supercharging Multimodal AutoML with Foundation Models](https://arxiv.org/pdf/2404.16233) (*AutoML Conf*, 2024) ([BibTeX](CITING.md#autogluonmultimodal))
+- [Multi-layer Stack Ensembles for Time Series Forecasting](https://arxiv.org/abs/2511.15350) (*AutoML Conf*, 2025) ([BibTeX](CITING.md#autogluontimeseries))
+- [Chronos-2: From Univariate to Universal Forecasting](https://arxiv.org/abs/2510.15821) (*Arxiv*, 2025) ([BibTeX](CITING.md#autogluontimeseries))
 
 ### Articles
 - [AutoGluon-TimeSeries: Every Time Series Forecasting Model In One Library](https://towardsdatascience.com/autogluon-timeseries-every-time-series-forecasting-model-in-one-library-29a3bf6879db) (*Towards Data Science*, Jan 2024)
@@ -232,5 +250,3 @@ We are actively accepting code contributions to the AutoGluon project. If you ar
 ## :classical_building: License
 
 This library is licensed under the Apache 2.0 License.
-
-