autogluon.tabular 1.2.1b20250225__py3-none-any.whl → 1.2.1b20250227__py3-none-any.whl

autogluon/tabular/learner/abstract_learner.py

@@ -39,19 +39,19 @@ class AbstractTabularLearner(AbstractLearner):
         self,
         path_context: str,
         label: str,
-        feature_generator: PipelineFeatureGenerator,
+        feature_generator: PipelineFeatureGenerator | None = None,
         ignored_columns: list = None,
-        label_count_threshold=10,
-        problem_type=None,
-        quantile_levels=None,
-        eval_metric=None,
-        positive_class=None,
-        cache_data=True,
-        is_trainer_present=False,
-        random_state=0,
-        sample_weight=None,
-        weight_evaluation=False,
-        groups=None,
+        label_count_threshold: int = 10,
+        problem_type: str | None = None,
+        quantile_levels: list[float] | None = None,
+        eval_metric: Scorer | None = None,
+        positive_class: str | None = None,
+        cache_data: bool = True,
+        is_trainer_present: bool = False,
+        random_state: int = 0,
+        sample_weight: str | None = None,
+        weight_evaluation: bool = False,
+        groups: str | None = None,
     ):
         super().__init__(path_context=path_context, random_state=random_state)
         self.label = label

autogluon/tabular/learner/default_learner.py

@@ -43,16 +43,17 @@ class DefaultLearner(AbstractTabularLearner):
     def _fit(
         self,
         X: DataFrame,
-        X_val: DataFrame = None,
-        X_test: DataFrame = None,
-        X_unlabeled: DataFrame = None,
-        holdout_frac=0.1,
-        num_bag_folds=0,
-        num_bag_sets=1,
-        time_limit=None,
-        infer_limit=None,
-        infer_limit_batch_size=None,
-        verbosity=2,
+        X_val: DataFrame | None = None,
+        X_test: DataFrame | None = None,
+        X_unlabeled: DataFrame | None = None,
+        holdout_frac: float = 0.1,
+        num_bag_folds: int = 0,
+        num_bag_sets: int = 1,
+        time_limit: float | None = None,
+        infer_limit: float | None = None,
+        infer_limit_batch_size: int | None = None,
+        verbosity: int = 2,
+        raise_on_model_failure: bool = False,
         **trainer_fit_kwargs,
     ):
         """Arguments:
@@ -121,6 +122,7 @@ class DefaultLearner(AbstractTabularLearner):
             save_data=self.cache_data,
             random_state=self.random_state,
             verbosity=verbosity,
+            raise_on_model_failure=raise_on_model_failure,
         )
 
         self.trainer_path = trainer.path

autogluon/tabular/models/automm/automm_model.py

@@ -26,6 +26,8 @@ logger = logging.getLogger(__name__)
 
 
 class MultiModalPredictorModel(AbstractModel):
+    ag_key = "AG_AUTOMM"
+    ag_name = "MultiModalPredictor"
     _NN_MODEL_NAME = "automm_model"
 
     def __init__(self, **kwargs):

autogluon/tabular/models/automm/ft_transformer.py

@@ -12,6 +12,9 @@ logger = logging.getLogger(__name__)
 
 # TODO: Add unit tests
 class FTTransformerModel(MultiModalPredictorModel):
+    ag_key = "FT_TRANSFORMER"
+    ag_name = "FTTransformer"
+
     def __init__(self, **kwargs):
         """Wrapper of autogluon.multimodal.MultiModalPredictor.
 

autogluon/tabular/models/catboost/catboost_model.py

@@ -2,6 +2,7 @@ import logging
 import math
 import os
 import time
+from types import MappingProxyType
 
 import numpy as np
 import pandas as pd
@@ -30,6 +31,12 @@ class CatBoostModel(AbstractModel):
 
     Hyperparameter options: https://catboost.ai/en/docs/references/training-parameters
     """
+    ag_key = "CAT"
+    ag_name = "CatBoost"
+    ag_priority = 70
+    ag_priority_by_problem_type = MappingProxyType({
+        SOFTCLASS: 60
+    })
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

autogluon/tabular/models/fastainn/tabular_nn_fastai.py

@@ -8,6 +8,7 @@ import warnings
 from builtins import classmethod
 from functools import partial
 from pathlib import Path
+from types import MappingProxyType
 from typing import Union
 
 import numpy as np
@@ -28,7 +29,7 @@ from autogluon.common.features.types import (
 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_fastai
-from autogluon.core.constants import BINARY, QUANTILE, REGRESSION
+from autogluon.core.constants import BINARY, MULTICLASS, QUANTILE, REGRESSION
 from autogluon.core.hpo.constants import RAY_BACKEND
 from autogluon.core.models import AbstractModel
 from autogluon.core.utils.exceptions import TimeLimitExceeded
@@ -92,6 +93,14 @@ class NNFastAiTabularModel(AbstractModel):
         'early.stopping.min_delta': 0.0001,
         'early.stopping.patience': 10,
     """
+    ag_key = "FASTAI"
+    ag_name = "NeuralNetFastAI"
+    ag_priority = 50
+    # Increase priority for multiclass since neural networks
+    # scale better than trees as a function of n_classes.
+    ag_priority_by_problem_type = MappingProxyType({
+        MULTICLASS: 95,
+    })
 
     model_internals_file_name = "model-internals.pkl"
 

autogluon/tabular/models/fasttext/fasttext_model.py

@@ -21,6 +21,9 @@ logger = logging.getLogger(__name__)
 
 
 class FastTextModel(AbstractModel):
+    ag_key = "FASTTEXT"
+    ag_name = "FastText"
+
     model_bin_file_name = "fasttext.ftz"
 
     def __init__(self, **kwargs):

autogluon/tabular/models/image_prediction/image_predictor.py

@@ -22,6 +22,8 @@ class ImagePredictorModel(MultiModalPredictorModel):
     Additionally has special null image handling to improve performance in the presence of null images (aka image path of '')
         Note: null handling has not been compared to the built-in null handling of MultimodalPredictor yet.
     """
+    ag_key = "AG_IMAGE_NN"
+    ag_name = "ImagePredictor"
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

autogluon/tabular/models/imodels/imodels_models.py

@@ -75,6 +75,9 @@ class _IModelsModel(AbstractModel):
 
 
 class RuleFitModel(_IModelsModel):
+    ag_key = "IM_RULEFIT"
+    ag_name = "RuleFit"
+
     def get_model(self):
         try_import_imodels()
         from imodels import RuleFitClassifier, RuleFitRegressor
@@ -86,6 +89,9 @@ class RuleFitModel(_IModelsModel):
 
 
 class GreedyTreeModel(_IModelsModel):
+    ag_key = "IM_GREEDYTREE"
+    ag_name = "GreedyTree"
+
     def get_model(self):
         try_import_imodels()
         from imodels import GreedyTreeClassifier
@@ -98,6 +104,9 @@ class GreedyTreeModel(_IModelsModel):
 
 
 class BoostedRulesModel(_IModelsModel):
+    ag_key = "IM_BOOSTEDRULES"
+    ag_name = "BoostedRules"
+
     def get_model(self):
         try_import_imodels()
         from imodels import BoostedRulesClassifier
@@ -109,6 +118,9 @@ class BoostedRulesModel(_IModelsModel):
 
 
 class HSTreeModel(_IModelsModel):
+    ag_key = "IM_HSTREE"
+    ag_name = "HierarchicalShrinkageTree"
+
     def get_model(self):
         try_import_imodels()
         from imodels import HSTreeClassifierCV, HSTreeRegressorCV
@@ -120,6 +132,9 @@ class HSTreeModel(_IModelsModel):
 
 
 class FigsModel(_IModelsModel):
+    ag_key = "IM_FIGS"
+    ag_name = "Figs"
+
     def get_model(self):
         try_import_imodels()
         from imodels import FIGSClassifier, FIGSRegressor

autogluon/tabular/models/knn/knn_model.py

@@ -22,6 +22,9 @@ class KNNModel(AbstractModel):
     """
     KNearestNeighbors model (scikit-learn): https://scikit-learn.org/stable/modules/generated/sklearn.neighbors.KNeighborsClassifier.html
     """
+    ag_key = "KNN"
+    ag_name = "KNeighbors"
+    ag_priority = 100
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

autogluon/tabular/models/lgb/lgb_model.py

@@ -7,6 +7,7 @@ import random
 import re
 import time
 import warnings
+from types import MappingProxyType
 
 import numpy as np
 import pandas as pd
@@ -40,6 +41,12 @@ class LGBModel(AbstractModel):
     Extra hyperparameter options:
         ag.early_stop : int, specifies the early stopping rounds. Defaults to an adaptive strategy. Recommended to keep default.
     """
+    ag_key = "GBM"
+    ag_name = "LightGBM"
+    ag_priority = 90
+    ag_priority_by_problem_type = MappingProxyType({
+        SOFTCLASS: 100
+    })
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

autogluon/tabular/models/lr/lr_model.py

@@ -38,6 +38,9 @@ class LinearModel(AbstractModel):
 
         'regression': https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.Ridge.html#sklearn.linear_model.Ridge
     """
+    ag_key = "LR"
+    ag_name = "LinearModel"
+    ag_priority = 30
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

autogluon/tabular/models/rf/rf_model.py

@@ -27,6 +27,9 @@ class RFModel(AbstractModel):
     """
     Random Forest model (scikit-learn): https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestClassifier.html
     """
+    ag_key = "RF"
+    ag_name = "RandomForest"
+    ag_priority = 80
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

autogluon/tabular/models/tab_transformer/tab_transformer_model.py

@@ -40,6 +40,8 @@ class TabTransformerModel(AbstractNeuralNetworkModel):
     and applies them to the use case of tabular data. Specifically, this makes TabTransformer suitable for unsupervised
     training of Tabular data with a subsequent fine-tuning step on labeled data.
     """
+    ag_key = "TRANSF"
+    ag_name = "Transformer"
 
     params_file_name = "tab_trans_params.pth"
 

autogluon/tabular/models/tabpfn/tabpfn_model.py

@@ -22,6 +22,9 @@ class TabPFNModel(AbstractModel):
     To use this model, `tabpfn` must be installed.
     To install TabPFN, you can run `pip install autogluon.tabular[tabpfn]` or `pip install tabpfn`.
     """
+    ag_key = "TABPFN"
+    ag_name = "TabPFN"
+    ag_priority = 110
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

autogluon/tabular/models/tabpfnmix/tabpfnmix_model.py

@@ -35,6 +35,10 @@ class TabPFNMixModel(AbstractModel):
 
     For more information, refer to the `./_internals/README.md` file.
     """
+    ag_key = "TABPFNMIX"
+    ag_name = "TabPFNMix"
+    ag_priority = 45
+
     weights_file_name = "model.pt"
 
     def __init__(self, **kwargs):

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

@@ -47,6 +47,9 @@ class TabularNeuralNetTorchModel(AbstractNeuralNetworkModel):
         ag.early_stop : int | str, default = "default"
             Specifies the early stopping rounds. Defaults to an adaptive strategy. Recommended to keep default.
     """
+    ag_key = "NN_TORCH"
+    ag_name = "NeuralNetTorch"
+    ag_priority = 25
 
     # Constants used throughout this class:
     unique_category_str = np.nan  # string used to represent missing values and unknown categories for categorical features.

autogluon/tabular/models/text_prediction/text_prediction_v1_model.py

@@ -19,6 +19,9 @@ logger = logging.getLogger(__name__)
 class TextPredictorModel(MultiModalPredictorModel):
     """MultimodalPredictor that doesn't use image features"""
 
+    ag_key = "AG_TEXT_NN"
+    ag_name = "TextPredictor"
+
     def _get_default_auxiliary_params(self) -> dict:
         default_auxiliary_params = super()._get_default_auxiliary_params()
         extra_auxiliary_params = dict(

autogluon/tabular/models/vowpalwabbit/vowpalwabbit_model.py

@@ -38,6 +38,9 @@ class VowpalWabbitModel(AbstractModel):
     VowpalWabbit Command Line args: https://github.com/VowpalWabbit/vowpal_wabbit/wiki/Command-line-arguments
 
     """
+    ag_key = "VW"
+    ag_name = "VowpalWabbit"
+    ag_priority = 10
 
     model_internals_file_name = "model-internals.pkl"
 

autogluon/tabular/models/xgboost/xgboost_model.py

@@ -27,6 +27,9 @@ class XGBoostModel(AbstractModel):
 
     Hyperparameter options: https://xgboost.readthedocs.io/en/latest/parameter.html
     """
+    ag_key = "XGB"
+    ag_name = "XGBoost"
+    ag_priority = 40
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)

autogluon/tabular/models/xt/xt_model.py

@@ -7,6 +7,9 @@ class XTModel(RFModel):
     """
     Extra Trees model (scikit-learn): https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.ExtraTreesClassifier.html#sklearn.ensemble.ExtraTreesClassifier
     """
+    ag_key = "XT"
+    ag_name = "ExtraTrees"
+    ag_priority = 60
 
     def _get_model_type(self):
         if self.problem_type == REGRESSION:

autogluon/tabular/predictor/predictor.py

@@ -990,6 +990,14 @@ class TabularPredictor:
                 to any amount of labeled data.
             verbosity : int
                 If specified, overrides the existing `predictor.verbosity` value.
+            raise_on_model_failure: bool, default = False
+                If True, will raise on any exception during model training.
+                    This is useful when using a debugger during development to identify the cause of model failures.
+                    This should only be used for debugging.
+                If False, will try to skip to the next model if an exception occurred during model training.
+                    This is the default logic and is a core principle of AutoGluon's design.
+
+                .. versionadded:: 1.3.0
             raise_on_no_models_fitted: bool, default = True
                 If True, will raise a RuntimeError if no models were successfully fit during `fit()`.
             calibrate: bool or str, default = 'auto'
@@ -1109,6 +1117,7 @@ class TabularPredictor:
         delay_bag_sets: bool = kwargs["delay_bag_sets"]
         test_data = kwargs["test_data"]
         learning_curves = kwargs["learning_curves"]
+        raise_on_model_failure = kwargs["raise_on_model_failure"]
 
         if ag_args is None:
             ag_args = {}
@@ -1256,6 +1265,7 @@ class TabularPredictor:
             verbosity=verbosity,
             use_bag_holdout=use_bag_holdout,
             callbacks=callbacks,
+            raise_on_model_failure=raise_on_model_failure,
         )
         ag_post_fit_kwargs = dict(
             keep_only_best=kwargs["keep_only_best"],
@@ -4330,7 +4340,14 @@ class TabularPredictor:
             reduce_children=reduce_children,
         )
 
-    def delete_models(self, models_to_keep=None, models_to_delete=None, allow_delete_cascade=False, delete_from_disk=True, dry_run=True):
+    def delete_models(
+        self,
+        models_to_keep: str | list[str] | None = None,
+        models_to_delete: str | list[str] | None = None,
+        allow_delete_cascade: bool = False,
+        delete_from_disk: bool = True,
+        dry_run: bool | None = None,
+    ):
         """
         Deletes models from `predictor`.
         This can be helpful to minimize memory usage and disk usage, particularly for model deployment.
@@ -4341,13 +4358,13 @@ class TabularPredictor:
 
         Parameters
         ----------
-        models_to_keep : str or list, default = None
+        models_to_keep : str or list[str], default = None
             Name of model or models to not delete.
             All models that are not specified and are also not required as a dependency of any model in `models_to_keep` will be deleted.
             Specify `models_to_keep='best'` to keep only the best model and its model dependencies.
             `models_to_delete` must be None if `models_to_keep` is set.
             To see the list of possible model names, use: `predictor.model_names()` or `predictor.leaderboard()`.
-        models_to_delete : str or list, default = None
+        models_to_delete : str or list[str], default = None
             Name of model or models to delete.
             All models that are not specified but depend on a model in `models_to_delete` will also be deleted.
             `models_to_keep` must be None if `models_to_delete` is set.
@@ -4361,10 +4378,19 @@ class TabularPredictor:
             WARNING: This deletes the entire directory for the deleted models, and ALL FILES located there.
                 It is highly recommended to first run with `dry_run=True` to understand which directories will be deleted.
         dry_run : bool, default = True
+            WARNING: Starting in v1.4.0 dry_run will default to False.
             If `True`, then deletions don't occur, and logging statements are printed describing what would have occurred.
             Set `dry_run=False` to perform the deletions.
 
         """
+        if dry_run is None:
+            warnings.warn(
+                f"dry_run was not specified for `TabularPredictor.delete_models`. dry_run prior to version 1.4.0 defaults to True. "
+                f"Starting in version 1.4, AutoGluon will default dry_run to False. "
+                f"If you want to maintain the current logic in future versions, explicitly specify `dry_run=True`.",
+                category=FutureWarning,
+            )
+            dry_run = True
         self._assert_is_fit("delete_models")
         if models_to_keep == "best":
             models_to_keep = self.model_best
@@ -5020,6 +5046,7 @@ class TabularPredictor:
             # learning curves and test data (for logging purposes only)
             learning_curves=False,
             test_data=None,
+            raise_on_model_failure=False,
         )
         kwargs, ds_valid_keys = self._sanitize_dynamic_stacking_kwargs(kwargs)
         kwargs = self._validate_fit_extra_kwargs(kwargs, extra_valid_keys=list(fit_kwargs_default.keys()) + ds_valid_keys)

autogluon/tabular/register/__init__.py

@@ -0,0 +1,2 @@
+from ._model_register import ModelRegister
+from ._ag_model_register import ag_model_register

autogluon/tabular/register/_ag_model_register.py

@@ -0,0 +1,66 @@
+from autogluon.core.models import (
+    DummyModel,
+    GreedyWeightedEnsembleModel,
+    SimpleWeightedEnsembleModel,
+)
+
+from . import ModelRegister
+from ..models import (
+    BoostedRulesModel,
+    CatBoostModel,
+    FastTextModel,
+    FigsModel,
+    FTTransformerModel,
+    GreedyTreeModel,
+    HSTreeModel,
+    ImagePredictorModel,
+    KNNModel,
+    LGBModel,
+    LinearModel,
+    MultiModalPredictorModel,
+    NNFastAiTabularModel,
+    RFModel,
+    RuleFitModel,
+    TabPFNMixModel,
+    TabPFNModel,
+    TabularNeuralNetTorchModel,
+    TextPredictorModel,
+    VowpalWabbitModel,
+    XGBoostModel,
+    XTModel,
+)
+from ..models.tab_transformer.tab_transformer_model import TabTransformerModel
+
+
+# When adding a new model officially to AutoGluon, the model class should be added to the bottom of this list.
+REGISTERED_MODEL_CLS_LST = [
+    RFModel,
+    XTModel,
+    KNNModel,
+    LGBModel,
+    CatBoostModel,
+    XGBoostModel,
+    TabularNeuralNetTorchModel,
+    LinearModel,
+    NNFastAiTabularModel,
+    TabTransformerModel,
+    TextPredictorModel,
+    ImagePredictorModel,
+    MultiModalPredictorModel,
+    FTTransformerModel,
+    TabPFNModel,
+    TabPFNMixModel,
+    FastTextModel,
+    VowpalWabbitModel,
+    GreedyWeightedEnsembleModel,
+    SimpleWeightedEnsembleModel,
+    RuleFitModel,
+    GreedyTreeModel,
+    FigsModel,
+    HSTreeModel,
+    BoostedRulesModel,
+    DummyModel,
+]
+
+# TODO: Replace logic in `autogluon.tabular.trainer.model_presets.presets` with `ag_model_register`
+ag_model_register = ModelRegister(model_cls_list=REGISTERED_MODEL_CLS_LST)

autogluon/tabular/register/_model_register.py

@@ -0,0 +1,146 @@
+from __future__ import annotations
+
+from typing import Type
+
+import pandas as pd
+
+from autogluon.core.models import AbstractModel
+
+
+# TODO: Move to core? Maybe TimeSeries can reuse?
+# TODO: Use this / refer to this in the custom model tutorial
+# TODO: Add to documentation website
+# TODO: Test register logic in AG
+class ModelRegister:
+    """
+    ModelRegister keeps track of all known model classes to AutoGluon.
+    It can provide information such as:
+        What model classes and keys are valid to specify in an AutoGluon predictor fit call.
+        What a model's name is.
+        What a model's key is (such as the key specified by the user in `hyperparameters` to refer to a specific model type).
+        What a model's priority is (aka which order to fit a list of models).
+
+    Additionally, users can register custom models to AutoGluon so the key is recognized in `hyperparameters` and is treated with the proper priority and name.
+    They can register new models via `ModelRegister.add(model_cls)`.
+
+    Therefore, if a user creates a custom model `MyCustomModel` that inherits from `AbstractModel`, they can set the class attributes in `MyCustomModel`:
+        ag_key: The string key that can be specified in `hyperparameters`. Example: "GBM" for LGBModel
+        ag_name: The string name that is used in logging and accessing the model. Example: "LightGBM" for LGBModel
+        ag_priority: The int priority that is used to order the fitting of models. Higher values will be fit before lower values. Default 0. Example: 90 for LGBModel
+        ag_priority_to_problem_type: A dictionary of problem_type to priority that overrides `ag_priority` if specified for a given problem_type. Optional.
+
+    Then they can say `ag_model_register.add(MyCustomModel)`.
+    Assuming MyCustomModel.ag_key = "MY_MODEL", they can now do:
+    ```
+    predictor.fit(..., hyperparameters={"MY_MODEL": ...})
+    ```
+    """
+    def __init__(self, model_cls_list: list[Type[AbstractModel]] | None = None):
+        if model_cls_list is None:
+            model_cls_list = []
+        assert isinstance(model_cls_list, list)
+        self._model_cls_list = []
+        self._key_to_cls_map = dict()
+        for model_cls in model_cls_list:
+            self.add(model_cls)
+
+    def exists(self, model_cls: Type[AbstractModel]) -> bool:
+        return model_cls in self._model_cls_list
+
+    def add(self, model_cls: Type[AbstractModel]):
+        """
+        Adds `model_cls` to the model register
+        """
+        assert not self.exists(model_cls), f"Cannot add model_cls that is already registered: {model_cls}"
+        if model_cls.ag_key is None:
+            raise AssertionError(
+                f"Cannot add model_cls with `ag_key=None`. "
+                f"Ensure you set class attribute `ag_key` to a string for your model_cls: {model_cls}"
+                f'\n\tFor example, LightGBModel sets `ag_key = "GBM"`'
+            )
+        if model_cls.ag_name is None:
+            raise AssertionError(
+                f"Cannot add model_cls with `ag_name=None`. "
+                f"Ensure you set class attribute `ag_name` to a string for your model_cls: {model_cls}"
+                f'\n\tFor example, LightGBModel sets `ag_name = "LightGBM"`'
+            )
+        assert isinstance(model_cls.ag_key, str)
+        assert isinstance(model_cls.ag_name, str)
+        assert isinstance(model_cls.ag_priority, int)
+        if model_cls.ag_key in self._key_to_cls_map:
+            raise AssertionError(
+                f"Cannot register a model class that shares a model key with an already registered model class."
+                f"\n`model_cls.ag_key` must be unique among registered models:"
+                f"\n\t        New  Class: {model_cls}"
+                f"\n\tConflicting  Class: {self._key_to_cls_map[model_cls.ag_key]}"
+                f"\n\tConflicting ag_key: {model_cls.ag_key}"
+            )
+        self._model_cls_list.append(model_cls)
+        self._key_to_cls_map[model_cls.ag_key] = model_cls
+
+    def remove(self, model_cls: Type[AbstractModel]):
+        """
+        Removes `model_cls` from the model register
+        """
+        assert self.exists(model_cls), f"Cannot remove model_cls that isn't registered: {model_cls}"
+        self._model_cls_list = [m for m in self._model_cls_list if m != model_cls]
+        self._key_to_cls_map.pop(model_cls.ag_key)
+
+    @property
+    def model_cls_list(self) -> list[Type[AbstractModel]]:
+        return self._model_cls_list
+
+    @property
+    def keys(self) -> list[str]:
+        return [self.key(model_cls) for model_cls in self.model_cls_list]
+
+    def key_to_cls_map(self) -> dict[str, Type[AbstractModel]]:
+        return self._key_to_cls_map
+
+    def key_to_cls(self, key: str) -> Type[AbstractModel]:
+        if key not in self._key_to_cls_map:
+            raise ValueError(
+                f"No registered model exists with provided key: {key}"
+                f"\n\tValid keys: {list(self.key_to_cls_map().keys())}"
+            )
+        return self.key_to_cls_map()[key]
+
+    def priority_map(self, problem_type: str | None = None) -> dict[Type[AbstractModel], int]:
+        return {model_cls: self.priority(model_cls, problem_type=problem_type) for model_cls in self._model_cls_list}
+
+    def key(self, model_cls: Type[AbstractModel]) -> str:
+        assert self.exists(model_cls), f"Model class must be registered: {model_cls}"
+        return model_cls.ag_key
+
+    def name_map(self) -> dict[Type[AbstractModel], str]:
+        return {model_cls: model_cls.ag_name for model_cls in self._model_cls_list}
+
+    def name(self, model_cls: Type[AbstractModel]) -> str:
+        assert self.exists(model_cls), f"Model class must be registered: {model_cls}"
+        return model_cls.ag_name
+
+    def priority(self, model_cls: Type[AbstractModel], problem_type: str | None = None) -> int:
+        assert self.exists(model_cls), f"Model class must be registered: {model_cls}"
+        return model_cls.get_ag_priority(problem_type=problem_type)
+
+    def docstring(self, model_cls: Type[AbstractModel]) -> str:
+        assert self.exists(model_cls), f"Model class must be registered: {model_cls}"
+        return model_cls.__doc__
+
+    # TODO: Could add a lot of information here to track which features are supported for each model:
+    #  ag.early_stop support
+    #  refit_full support
+    #  GPU support
+    #  etc.
+    def to_frame(self) -> pd.DataFrame:
+        model_classes = self.model_cls_list
+        cls_dict = {}
+        for model_cls in model_classes:
+            cls_dict[self.key(model_cls)] = {
+                "model_cls": model_cls.__name__,
+                "ag_name": self.name(model_cls),
+                "ag_priority": self.priority(model_cls),
+            }
+        df = pd.DataFrame(cls_dict).T
+        df.index.name = "ag_key"
+        return df

autogluon/tabular/trainer/abstract_trainer.py

@@ -128,6 +128,12 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
         Higher levels correspond to more detailed print statements (you can set verbosity = 0 to suppress warnings).
         If using logging, you can alternatively control amount of information printed via `logger.setLevel(L)`,
         where `L` ranges from 0 to 50 (Note: higher values of `L` correspond to fewer print statements, opposite of verbosity levels).
+    raise_on_model_failure : bool, default = False
+        If True, Trainer will raise on any exception during model training.
+            This is ideal when using a debugger during development.
+        If False, Trainer will try to skip to the next model if an exception occurred during model training.
+
+        .. versionadded:: 1.3.0
     """
 
     distill_stackname = "distill"  # name of stack-level for distilled student models
@@ -149,6 +155,7 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
         save_data: bool = False,
         random_state: int = 0,
         verbosity: int = 2,
+        raise_on_model_failure: bool = False,
     ):
         super().__init__(
             path=path,
@@ -163,6 +170,7 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
         #: Integer value added to the stack level to get the random_state for kfold splits or the train/val split if bagging is disabled
         self.random_state = random_state
         self.verbosity = verbosity
+        self.raise_on_model_failure = raise_on_model_failure
 
         # TODO: consider redesign where Trainer doesn't need sample_weight column name and weights are separate from X
         self.sample_weight = sample_weight
@@ -2188,6 +2196,10 @@ class AbstractTabularTrainer(AbstractTrainer[AbstractModel]):
             # TODO: Add recursive=True to avoid repeatedly loading models each time this is called for bagged ensembles (especially during repeated bagging)
             self.save_model(model=model)
         except Exception as exc:
+            if self.raise_on_model_failure:
+                # immediately raise instead of skipping to next model, useful for debugging during development
+                logger.warning("Model failure occurred... Raising exception instead of continuing to next model. (raise_on_model_failure=True)")
+                raise exc
             exception = exc  # required to reference exc outside of `except` statement
             del_model = True
             if isinstance(exception, TimeLimitExceeded):

autogluon/tabular/trainer/model_presets/presets.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import copy
 import inspect
 import logging
@@ -22,70 +24,16 @@ from autogluon.core.constants import (
 )
 from autogluon.core.models import (
     AbstractModel,
-    DummyModel,
-    GreedyWeightedEnsembleModel,
-    SimpleWeightedEnsembleModel,
     StackerEnsembleModel,
 )
 from autogluon.core.trainer.utils import process_hyperparameters
 
-from ...models import (
-    BoostedRulesModel,
-    CatBoostModel,
-    FastTextModel,
-    FigsModel,
-    FTTransformerModel,
-    GreedyTreeModel,
-    HSTreeModel,
-    ImagePredictorModel,
-    KNNModel,
-    LGBModel,
-    LinearModel,
-    MultiModalPredictorModel,
-    NNFastAiTabularModel,
-    RFModel,
-    RuleFitModel,
-    TabPFNMixModel,
-    TabPFNModel,
-    TabularNeuralNetTorchModel,
-    TextPredictorModel,
-    VowpalWabbitModel,
-    XGBoostModel,
-    XTModel,
-)
-from ...models.tab_transformer.tab_transformer_model import TabTransformerModel
+from ...register import ag_model_register
 from ...version import __version__
 
 logger = logging.getLogger(__name__)
 
-# Higher values indicate higher priority, priority dictates the order models are trained for a given level.
-DEFAULT_MODEL_PRIORITY = dict(
-    TABPFN=110,  # highest priority due to its very fast training time
-    KNN=100,
-    GBM=90,
-    RF=80,
-    CAT=70,
-    XT=60,
-    FASTAI=50,
-    TABPFNMIX=45,
-    XGB=40,
-    LR=30,
-    NN_TORCH=25,
-    VW=10,
-    FASTTEXT=0,
-    AG_TEXT_NN=0,
-    AG_IMAGE_NN=0,
-    AG_AUTOMM=0,
-    TRANSF=0,
-    custom=0,
-    # interpretable models
-    IM_RULEFIT=0,
-    IM_GREEDYTREE=0,
-    IM_FIGS=0,
-    IM_HSTREE=0,
-    IM_BOOSTEDRULES=0,
-)
-
+# TODO: Replace with ag_model_register
 # Problem type specific model priority overrides (will update default values in DEFAULT_MODEL_PRIORITY)
 PROBLEM_TYPE_MODEL_PRIORITY = {
     MULTICLASS: dict(
@@ -93,6 +41,7 @@ PROBLEM_TYPE_MODEL_PRIORITY = {
     ),
 }
 
+# TODO: Replace with ag_model_register
 DEFAULT_SOFTCLASS_PRIORITY = dict(
     GBM=100,
     RF=80,
@@ -102,66 +51,11 @@ DEFAULT_SOFTCLASS_PRIORITY = dict(
 
 DEFAULT_CUSTOM_MODEL_PRIORITY = 0
 
-MODEL_TYPES = dict(
-    RF=RFModel,
-    XT=XTModel,
-    KNN=KNNModel,
-    GBM=LGBModel,
-    CAT=CatBoostModel,
-    XGB=XGBoostModel,
-    NN_TORCH=TabularNeuralNetTorchModel,
-    LR=LinearModel,
-    FASTAI=NNFastAiTabularModel,
-    TRANSF=TabTransformerModel,
-    AG_TEXT_NN=TextPredictorModel,
-    AG_IMAGE_NN=ImagePredictorModel,
-    AG_AUTOMM=MultiModalPredictorModel,
-    FT_TRANSFORMER=FTTransformerModel,
-    TABPFN=TabPFNModel,
-    TABPFNMIX=TabPFNMixModel,
-    FASTTEXT=FastTextModel,
-    ENS_WEIGHTED=GreedyWeightedEnsembleModel,
-    SIMPLE_ENS_WEIGHTED=SimpleWeightedEnsembleModel,
-    # interpretable models
-    IM_RULEFIT=RuleFitModel,
-    IM_GREEDYTREE=GreedyTreeModel,
-    IM_FIGS=FigsModel,
-    IM_HSTREE=HSTreeModel,
-    IM_BOOSTEDRULES=BoostedRulesModel,
-    VW=VowpalWabbitModel,
-    DUMMY=DummyModel,
-)
-
-
-# TODO: v1.0 Have this be defined in the model class
-DEFAULT_MODEL_NAMES = {
-    RFModel: "RandomForest",
-    XTModel: "ExtraTrees",
-    KNNModel: "KNeighbors",
-    LGBModel: "LightGBM",
-    CatBoostModel: "CatBoost",
-    XGBoostModel: "XGBoost",
-    TabularNeuralNetTorchModel: "NeuralNetTorch",
-    LinearModel: "LinearModel",
-    NNFastAiTabularModel: "NeuralNetFastAI",
-    TabTransformerModel: "Transformer",
-    TextPredictorModel: "TextPredictor",
-    ImagePredictorModel: "ImagePredictor",
-    MultiModalPredictorModel: "MultiModalPredictor",
-    FTTransformerModel: "FTTransformer",
-    TabPFNModel: "TabPFN",
-    TabPFNMixModel: "TabPFNMix",
-    FastTextModel: "FastText",
-    VowpalWabbitModel: "VowpalWabbit",
-    GreedyWeightedEnsembleModel: "WeightedEnsemble",
-    SimpleWeightedEnsembleModel: "WeightedEnsemble",
-    # Interpretable models
-    RuleFitModel: "RuleFit",
-    GreedyTreeModel: "GreedyTree",
-    FigsModel: "Figs",
-    HSTreeModel: "HierarchicalShrinkageTree",
-    BoostedRulesModel: "BoostedRules",
-}
+# FIXME: Don't do this, use ag_model_register lazily so users can register custom models before calling fit
+DEFAULT_MODEL_PRIORITY = {ag_model_register.key(model_cls): ag_model_register.priority(model_cls) for model_cls in ag_model_register.model_cls_list}
+DEFAULT_MODEL_NAMES = ag_model_register.name_map()
+REGISTERED_MODEL_CLS_LST = ag_model_register.model_cls_list
+MODEL_TYPES = ag_model_register.key_to_cls_map()
 
 
 VALID_AG_ARGS_KEYS = {

autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.2.1b20250225"
+__version__ = "1.2.1b20250227"
 __lite__ = False

{autogluon.tabular-1.2.1b20250225.dist-info → autogluon.tabular-1.2.1b20250227.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.2.1b20250225
+Version: 1.2.1b20250227
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -41,19 +41,19 @@ Requires-Dist: scipy<1.16,>=1.5.4
 Requires-Dist: pandas<2.3.0,>=2.0.0
 Requires-Dist: scikit-learn<1.5.3,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.2.1b20250225
-Requires-Dist: autogluon.features==1.2.1b20250225
+Requires-Dist: autogluon.core==1.2.1b20250227
+Requires-Dist: autogluon.features==1.2.1b20250227
 Provides-Extra: all
-Requires-Dist: torch<2.6,>=2.2; extra == "all"
-Requires-Dist: spacy<3.8; extra == "all"
-Requires-Dist: huggingface-hub[torch]; extra == "all"
-Requires-Dist: lightgbm<4.6,>=4.0; extra == "all"
-Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: autogluon.core[all]==1.2.1b20250225; extra == "all"
-Requires-Dist: catboost<1.3,>=1.2; extra == "all"
-Requires-Dist: xgboost<2.2,>=1.6; extra == "all"
 Requires-Dist: numpy<2.0.0,>=1.25; extra == "all"
 Requires-Dist: fastai<2.8,>=2.3.1; extra == "all"
+Requires-Dist: xgboost<2.2,>=1.6; extra == "all"
+Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: einops<0.9,>=0.7; extra == "all"
+Requires-Dist: lightgbm<4.6,>=4.0; extra == "all"
+Requires-Dist: huggingface-hub[torch]; extra == "all"
+Requires-Dist: torch<2.6,>=2.2; extra == "all"
+Requires-Dist: spacy<3.8; extra == "all"
+Requires-Dist: autogluon.core[all]==1.2.1b20250227; extra == "all"
 Provides-Extra: catboost
 Requires-Dist: numpy<2.0.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
@@ -66,7 +66,7 @@ Requires-Dist: imodels<1.4.0,>=1.3.10; extra == "imodels"
 Provides-Extra: lightgbm
 Requires-Dist: lightgbm<4.6,>=4.0; extra == "lightgbm"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.2.1b20250225; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.2.1b20250227; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.1,>=2024.0; extra == "skex"
 Provides-Extra: skl2onnx

{autogluon.tabular-1.2.1b20250225.dist-info → autogluon.tabular-1.2.1b20250227.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.2.1b20250225-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.2.1b20250227-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=_BxegzmyRYnkIA1LhdhXXgU3yeZs50K9F3dits4gPdc,91
+autogluon/tabular/version.py,sha256=5ozpv_b_rM463xTXRdXij1xDVDTJr5oIBlgIcEiF2lw,91
 autogluon/tabular/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/configs/config_helper.py,sha256=Pb2aW9Z9w77pYKPRVZ3nBzHY3KJaiEJSJ747zZcJIVk,21132
 autogluon/tabular/configs/feature_generator_presets.py,sha256=EV5Ym8VW15q92MwOUpTi7wZFS2QooM51fLg3RdUsn-M,1223
@@ -14,18 +14,18 @@ autogluon/tabular/experimental/_tabular_classifier.py,sha256=7lGoFdvkHiZS3VpcXo9
 autogluon/tabular/experimental/_tabular_regressor.py,sha256=EzEDL-19T5QUVNmLkSHNzzGwYrUxyqlNpIDPMgtV6Gg,1932
 autogluon/tabular/experimental/plot_leaderboard.py,sha256=BN_kB-zmOZNUYWyI7z9pF67GCV20zo8yV51HKKj1SCY,9481
 autogluon/tabular/learner/__init__.py,sha256=Hhmk5WpKQHohVmI-veOaKMelKJpIdzeXrmw_DPn3DTU,63
-autogluon/tabular/learner/abstract_learner.py,sha256=3myDh867x-EWTPR-O-iw82WGgd5n1NKWf3kaTcYQeh0,54955
-autogluon/tabular/learner/default_learner.py,sha256=cg3K0oA-4ccXWtmGgy6qUJaLldieFwDvnMP_PyE9gdk,24579
+autogluon/tabular/learner/abstract_learner.py,sha256=HmnW7KO3sV4H1QSquJn9DYOwoWa2vohKThq_hr4OHM4,55102
+autogluon/tabular/learner/default_learner.py,sha256=hjdKbcFtIQxQ3-k1LiGOo-w5sLxIIQAyFLs3-R35aw0,24781
 autogluon/tabular/models/__init__.py,sha256=tDVqwVG9q2ctLWRouyXeYs5NiSBnOnwh3anAfZyM3jg,1099
 autogluon/tabular/models/_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/_utils/rapids_utils.py,sha256=gbej9Hjn4alCWZuGN9sOLXMMAyWbgHPThTsp2feS39o,1038
 autogluon/tabular/models/_utils/torch_utils.py,sha256=dxs_KMMAOmNkRNjYf_hrzqaHIfkqn1xoKRKqCFbQ1Rk,537
 autogluon/tabular/models/automm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/automm/automm_model.py,sha256=z16uBs9SAGxuCDbROhTNOTzc4KhCd-ejX31Mt_FHNHY,11159
-autogluon/tabular/models/automm/ft_transformer.py,sha256=OxD5K8xZFxt8PutcGmjaZsAO2QoPpdDG0YoaxnTqrV0,3840
+autogluon/tabular/models/automm/automm_model.py,sha256=_FPkXiHjowuc_KTKVkrzstaj1pecPDxIAZfw0LVjbJ0,11220
+autogluon/tabular/models/automm/ft_transformer.py,sha256=8T80u3SbWr2-0KagOXhYpZkTvFIbRF91rE8sVZN0w_0,3901
 autogluon/tabular/models/catboost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/catboost/callbacks.py,sha256=l8x17n_w7oEFs-iDECSdBKZ89yW5g1z-zvj4XLgQPkw,7098
-autogluon/tabular/models/catboost/catboost_model.py,sha256=mURn4GGGXgKovsctXLw3ZPva9xVS_grIm2Lcrj0CdSk,17043
+autogluon/tabular/models/catboost/catboost_model.py,sha256=lzHsaY-EuihoSjZ7owVqAfzmdT5VVlMTQeIZ8_PJcxY,17225
 autogluon/tabular/models/catboost/catboost_softclass_utils.py,sha256=UiW0SUb3hFueW5qYtQn6Sbk7Wg7BWN4jqKWeFtbMvgU,3919
 autogluon/tabular/models/catboost/catboost_utils.py,sha256=YSc94V4DjrwbmkeUM8306zV7z21oq-K-qGCOj0UE_wg,3167
 autogluon/tabular/models/catboost/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -36,39 +36,39 @@ autogluon/tabular/models/fastainn/callbacks.py,sha256=3WvOEwqd1YAVInooKsFOTzAkCL
 autogluon/tabular/models/fastainn/fastai_helpers.py,sha256=gGYzyrAFl8hi8GnsemZNLGZn5xr7cyJXdFl08PIlza4,1393
 autogluon/tabular/models/fastainn/imports_helper.py,sha256=ICxA8ty47-oZu0Q9AjKCQe8uVi340Iu0NFruxvJPrbA,330
 autogluon/tabular/models/fastainn/quantile_helpers.py,sha256=d89GKvSRBgOy9EqcDI83MK5sqPRxP6JJ3BmPLmKnB0o,1808
-autogluon/tabular/models/fastainn/tabular_nn_fastai.py,sha256=ZGHpa6FY-3NxDUVQk5fTBeiIt735xEGUbuPgELxij_8,28957
+autogluon/tabular/models/fastainn/tabular_nn_fastai.py,sha256=SsBtfYRFuBXrN7LaCaJqLSLd1ZLVad5SXtHa_pxdG1w,29282
 autogluon/tabular/models/fastainn/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/fastainn/hyperparameters/parameters.py,sha256=DkQwAZZ7CuODKoljr-yrkx-uFxBSPRxkKuvPdwO-UhQ,2069
 autogluon/tabular/models/fastainn/hyperparameters/searchspaces.py,sha256=5qdknZDrHtdPdrhSqjamYQrCxvupXvlN3bVGEPgs48E,1660
 autogluon/tabular/models/fasttext/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/fasttext/fasttext_model.py,sha256=MKB5OQIfnezg-Co9KDHuXJICCD2sdcBqOStJP0JnevY,7040
+autogluon/tabular/models/fasttext/fasttext_model.py,sha256=0_Dbjz0_rzrRtBKVALlGZ8khnqeO5hPsvmZHi_ktv78,7090
 autogluon/tabular/models/fasttext/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/fasttext/hyperparameters/parameters.py,sha256=DbkLlHlxRh1uGWJ_sUYNrweSJj4yjlOBH_H2COyaWL8,1234
 autogluon/tabular/models/image_prediction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/image_prediction/image_predictor.py,sha256=ZRaykOm7XSlpXiX-ZmTRq5c8IZKspL7sRjLHmJswwUk,5396
+autogluon/tabular/models/image_prediction/image_predictor.py,sha256=2KBxnFzw-wAIjKYpJkmwKfsB_laUpWbY5HOxyDL6uqQ,5454
 autogluon/tabular/models/imodels/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/imodels/imodels_models.py,sha256=R7RuN6meg_SAkW7D59LnLusb_nyE4kiW3y4L_WXyj1s,4430
+autogluon/tabular/models/imodels/imodels_models.py,sha256=JE-VFeFc8Ot2T1IUtlNMP4_T-9F1I1NzQt4zA5aZ-to,4712
 autogluon/tabular/models/knn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/knn/_knn_loo_variants.py,sha256=-n2znYS7OBA0bZvtei6JZiEMRWp4GX-Qp64uheaHyhQ,4562
-autogluon/tabular/models/knn/knn_model.py,sha256=6O9ELnWN5L2MP8PDZCPafRBMxSHDIk6w13WBdNc87Eg,13806
+autogluon/tabular/models/knn/knn_model.py,sha256=fN8ybCC3zb9aVXS18M41RS78u58bvEvrv3ImELY_WOg,13874
 autogluon/tabular/models/knn/knn_rapids_model.py,sha256=0FFApNZFH8nyrDqlBSUV7jO-2fLe0-h_UHp1GsyQJ8E,1550
 autogluon/tabular/models/knn/knn_utils.py,sha256=XU1cxVXp1BAoQnja2_KmSIn9_q9gZkjAya7-9b0uStk,7455
 autogluon/tabular/models/lgb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/lgb/callbacks.py,sha256=0X42-nAbftKnu_zmFPDf8S3RrUJJjsJ1Qs_TPAJxzjU,11367
-autogluon/tabular/models/lgb/lgb_model.py,sha256=udmA6A-2RhVLRcaJXbjZCWrOhansButwRPRO7u0-BnM,24715
+autogluon/tabular/models/lgb/lgb_model.py,sha256=BKS9zMfj65Xnp74IK2IU_i9RfY4dWmiJxGqp7IiyWCQ,24898
 autogluon/tabular/models/lgb/lgb_utils.py,sha256=jzTDTzP-z7gcBGZyy1_0YkyTOLbU5DLeRqtil4FCZPI,7382
 autogluon/tabular/models/lgb/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/lgb/hyperparameters/parameters.py,sha256=LLEQ-Ns3HElWBsFJx3ogRV7L6qw_nXlcl7EyO0C0fVQ,1336
 autogluon/tabular/models/lgb/hyperparameters/searchspaces.py,sha256=tvNNR7niWz_B-PndYQXb6vVNABxSfBYRHj6ZVQJ1x2E,1930
 autogluon/tabular/models/lr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/lr/lr_model.py,sha256=LdAeMVcghKUZUacQbbHyCU7O8GDeg7pTye6Sb41YLxQ,15366
+autogluon/tabular/models/lr/lr_model.py,sha256=t2rZJf47bGJ5YlGrg100-LpAhNIPXE2KwaxDTyHxgSs,15433
 autogluon/tabular/models/lr/lr_preprocessing_utils.py,sha256=zkmVZtv05BQPDasVBz1J8LmXEfLgoggsv57s6cXuTMQ,1094
 autogluon/tabular/models/lr/lr_rapids_model.py,sha256=a07JvjWemrL0L08moA3K4lnYieukRlAdb2Z_uWA44k8,2127
 autogluon/tabular/models/lr/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/lr/hyperparameters/parameters.py,sha256=Hr5YC13zjbt3CfCbzGj8iXUIuDn-Q7FvDT2uSuiSVlM,1414
 autogluon/tabular/models/lr/hyperparameters/searchspaces.py,sha256=Igywc-B6qJ9EBLdasrDhW-Ot5FGirIzbXLwv5HRe5Xo,276
 autogluon/tabular/models/rf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/rf/rf_model.py,sha256=g6WTkJhJReCTXuhOmcDrb0qWum7-n2i7xivAx6l06Vo,21274
+autogluon/tabular/models/rf/rf_model.py,sha256=U9prO0B3Uk7Ycvagau7rgm_fNkBaJS1Km40iY9XMa3I,21342
 autogluon/tabular/models/rf/rf_quantile.py,sha256=2S8FE8po9lMnZaeKuVkzOUFOcdil46ZbFqm49OuvNZY,36460
 autogluon/tabular/models/rf/rf_rapids_model.py,sha256=3s-8M11dzCl_2Lu5iB3H8YjHLgyP_SElrm_4w_HfmqY,2028
 autogluon/tabular/models/rf/compilers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -80,15 +80,15 @@ autogluon/tabular/models/tab_transformer/pretexts.py,sha256=UEoDq_8hLKbY7EbE5IyL
 autogluon/tabular/models/tab_transformer/tab_model_base.py,sha256=4rmY1IrwoFuJejy-9gOoYSz-ar3DvZY8uXyDUBKk7Iw,3615
 autogluon/tabular/models/tab_transformer/tab_transformer.py,sha256=1c1oTJfSsGxQjzZJVN8doqFmYV-Wwwbqcu7RcW77kJk,6991
 autogluon/tabular/models/tab_transformer/tab_transformer_encoder.py,sha256=v2G1S_MSESzKqtvSfxS5uEse2CWtOn_K2E-uIwuE6zI,24701
-autogluon/tabular/models/tab_transformer/tab_transformer_model.py,sha256=pwqoig6SL3PutYhsB1o3mFyvBubXm91ynPGujp3Ab3g,22702
+autogluon/tabular/models/tab_transformer/tab_transformer_model.py,sha256=E_qPm2AMZY3I7EdKqPD5APKDbat5S-A7ohKaXAO-29w,22752
 autogluon/tabular/models/tab_transformer/utils.py,sha256=rrNk0X6Y0vzt7ivylRmTf0zjDx1DIF-5_ibf6B9Taz8,4554
 autogluon/tabular/models/tab_transformer/hyperparameters/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 autogluon/tabular/models/tab_transformer/hyperparameters/parameters.py,sha256=-vJRG8PVj5FgQnF9FNJHMvoIzzyazGE4XLRyQKL5VT8,3854
 autogluon/tabular/models/tab_transformer/hyperparameters/searchspaces.py,sha256=poiOFwOVIf1ONcPIjOqsA31YbqBgWxy0DlVFpVwKNHM,650
 autogluon/tabular/models/tabpfn/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon/tabular/models/tabpfn/tabpfn_model.py,sha256=v0S1jQjsws2gWX6Fu0vXqExmLhV6VWwljr5dHHvSKR8,6850
+autogluon/tabular/models/tabpfn/tabpfn_model.py,sha256=xdxBq9Bg4e8Cg6Y7jfUEwWr60mM7OvSOOLN0usrUn0I,6917
 autogluon/tabular/models/tabpfnmix/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabpfnmix/tabpfnmix_model.py,sha256=2XVYObbmi30FKPXrszfKlV34G1JnSiSikFk1Mn5Lq9M,16692
+autogluon/tabular/models/tabpfnmix/tabpfnmix_model.py,sha256=mc3mMRwytBYaxXjnnrlAQyS2dowT9sRd4xfvEkJfCW8,16765
 autogluon/tabular/models/tabpfnmix/_internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabpfnmix/_internal/tabpfnmix_classifier.py,sha256=U3DAhtSLpHUu-22GgR3QCZJnpRYpOW95XoTV7tE9J5Y,3425
 autogluon/tabular/models/tabpfnmix/_internal/tabpfnmix_regressor.py,sha256=J6JvrK6L6y3s-Ah6sHQdjSK0mwAMP-Wy3RRBwzB0AoA,3196
@@ -121,7 +121,7 @@ autogluon/tabular/models/tabular_nn/hyperparameters/__init__.py,sha256=47DEQpj8H
 autogluon/tabular/models/tabular_nn/hyperparameters/parameters.py,sha256=Z3t_U1f7jfolPey6lzqgJyoFbVgoncFNSvCKXSuLxeU,6465
 autogluon/tabular/models/tabular_nn/hyperparameters/searchspaces.py,sha256=pT9cJ3MaWPnaQwAf47Yz6f0-L9qDBknahERbggAp52U,2810
 autogluon/tabular/models/tabular_nn/torch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py,sha256=1ZJHM52wMZHSbXty19ykKA5tNql6ISO7nuFmAAuR-uU,42324
+autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py,sha256=QxZZ9XILnO4VdZ3avNNaPV7K16OsINq4SAWN9W9O3o4,42400
 autogluon/tabular/models/tabular_nn/torch/tabular_torch_dataset.py,sha256=RdnQGZSrvY1iuJB4JTANniH3Dorw-DP0Em_JK3_h7RM,13497
 autogluon/tabular/models/tabular_nn/torch/torch_network_modules.py,sha256=Qc3PwXTD8A7PgXi6EGuaBCrN3jsFAXDLCW7i6tE5wYI,11338
 autogluon/tabular/models/tabular_nn/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -129,35 +129,38 @@ autogluon/tabular/models/tabular_nn/utils/categorical_encoders.py,sha256=uLQaHku
 autogluon/tabular/models/tabular_nn/utils/data_preprocessor.py,sha256=ypXqtxdt1qH6la1hcq-BJ0dzQBNtgKY-BjXmIWxPjCg,5237
 autogluon/tabular/models/tabular_nn/utils/nn_architecture_utils.py,sha256=tttzR5EtYcFa6sIrUG9wyegdYmYE5DPK_CiLF1-L3c8,2875
 autogluon/tabular/models/text_prediction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/text_prediction/text_prediction_v1_model.py,sha256=PBrPyxE4rfXkQv99eHsTd4_Ccn-DPAtwg-jMjNBuYRk,867
+autogluon/tabular/models/text_prediction/text_prediction_v1_model.py,sha256=XCKKb9N3mKXLtbMHMjQ3y5HwKQFF25enA287WHv-ebc,924
 autogluon/tabular/models/vowpalwabbit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/vowpalwabbit/vowpalwabbit_model.py,sha256=UViJnoDBTP9Msn19JR-ri1ewp_3n2wZdJecJZsW046M,11631
+autogluon/tabular/models/vowpalwabbit/vowpalwabbit_model.py,sha256=hToCVe0KP3PIkOl3-jEh0R1m7_9xwGXMY8RRH2bzXnM,11699
 autogluon/tabular/models/vowpalwabbit/vowpalwabbit_utils.py,sha256=jZ0STjvqwKw8jJDeoo5yAXTvgwFvY8Fsz6OqSif_JGI,3677
 autogluon/tabular/models/xgboost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/xgboost/callbacks.py,sha256=uynimXya07XQMBkDvec-7mXK6OfMGP6M8MiVYu8OVRI,7008
-autogluon/tabular/models/xgboost/xgboost_model.py,sha256=_K-4RrshJtMkByvsuJYEOxzP9yQTILFsw8nQiDiH-b0,14258
+autogluon/tabular/models/xgboost/xgboost_model.py,sha256=4hMABSzg3CJnpWwiS3A7qhzWpNcC6BWyVAoC2MMr-TI,14322
 autogluon/tabular/models/xgboost/xgboost_utils.py,sha256=FVqZ8h4JAe_pifSvNx83cLZHwsuzTXylrrcan07AoNo,5757
 autogluon/tabular/models/xgboost/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/xgboost/hyperparameters/parameters.py,sha256=ay6bVVpiPzftbtz6TTS76w7j4vjDjzHFpuf2Bjf6Zu4,1673
 autogluon/tabular/models/xgboost/hyperparameters/searchspaces.py,sha256=lFwI34pcRtlVQkxmsdZsSaPry8t_WSMBhig4soMK54k,2140
 autogluon/tabular/models/xt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/xt/xt_model.py,sha256=gimCfMeTmhAJTy4ekCI_dbK1FluJ7EQMoHt0pGMvVgs,759
+autogluon/tabular/models/xt/xt_model.py,sha256=z9Hcp3CR_QS4XMlE9-g5rFknj2KPHfFLtyqb0qvOYbw,825
 autogluon/tabular/predictor/__init__.py,sha256=zCMgjxQlWpDWnr1l1xjBCiK3rWC3N3RoD8UXBnazT74,107
 autogluon/tabular/predictor/interpretable_predictor.py,sha256=5UeKgnMFsfY65tiO3kxfHBPr03lyswLrgdtjPhI0Y7Q,6934
-autogluon/tabular/predictor/predictor.py,sha256=nJu66FgSjjKUrHmflj90eq0SlPfVluz-uI84E7zHL-k,355670
+autogluon/tabular/predictor/predictor.py,sha256=A86C_9ixByrSrcU-Bxy9eTTZ7nCnIvRDN2uY-0ANLVY,357082
+autogluon/tabular/register/__init__.py,sha256=7CLOTWIUho0wi4eAwhYJ5Y0PfvNCWKnRwlw3bwYoTNE,93
+autogluon/tabular/register/_ag_model_register.py,sha256=tMr-QgxgCE49tdThdSFOZaJg2D9ckDh6fiR5K4cRtvk,1564
+autogluon/tabular/register/_model_register.py,sha256=jqSg0d89dXAAcp-OT4II90ce994ByKMMzAYmpkyaRbI,6824
 autogluon/tabular/trainer/__init__.py,sha256=PW_PGL-tWoQzx3ES2S53bQEZOtsRWTYiM9QdOqsk0dI,38
-autogluon/tabular/trainer/abstract_trainer.py,sha256=daz6_IfYX9bvQ5c5RpbxZ9f5JytUDq2-XDO1wAXpXtk,231155
+autogluon/tabular/trainer/abstract_trainer.py,sha256=lqOjVTLUaZNue4B7u47PYXTXsBEFbSJ4SyruNeChFCk,231925
 autogluon/tabular/trainer/auto_trainer.py,sha256=FyRWM8iUJuDvw_aqV5EV_xdh_pb-nHzAvG1sbEhvs0g,8680
 autogluon/tabular/trainer/model_presets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/trainer/model_presets/presets.py,sha256=1E-Z1FxUpyydaoEdxcTCg7bvSwVEuwImhH2-VlcRdaQ,19217
+autogluon/tabular/trainer/model_presets/presets.py,sha256=bTPGPyz07a7GG6327yO6ryuWbNc1aq3hF1qzZL-Xe4c,16733
 autogluon/tabular/trainer/model_presets/presets_distill.py,sha256=MnFC2GJc6RmDBNAGbsO2XMfo3PjR8cUrZoilWW8gTYQ,3295
 autogluon/tabular/tuning/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/tuning/feature_pruner.py,sha256=9iNku8gVbYEkjuKlyITPJDicsNkoraaQOlINQq9iZlQ,6877
-autogluon.tabular-1.2.1b20250225.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.2.1b20250225.dist-info/METADATA,sha256=ntal9IyVU1Qf94AAwh2wVQmmbT21HB3HCKfspoc6e9c,14386
-autogluon.tabular-1.2.1b20250225.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.2.1b20250225.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.2.1b20250225.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250225.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250225.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.2.1b20250225.dist-info/RECORD,,
+autogluon.tabular-1.2.1b20250227.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.2.1b20250227.dist-info/METADATA,sha256=nw0R8W2F-rUL_zr6eevigQGHEmmFKBhH2TGfPHvjuUE,14386
+autogluon.tabular-1.2.1b20250227.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.2.1b20250227.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.2.1b20250227.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250227.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250227.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.2.1b20250227.dist-info/RECORD,,