autogluon.tabular 1.2.1b20250218__tar.gz → 1.2.1b20250220__tar.gz

{autogluon.tabular-1.2.1b20250218 → autogluon.tabular-1.2.1b20250220}/PKG-INFO

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

{autogluon.tabular-1.2.1b20250218 → autogluon.tabular-1.2.1b20250220}/src/autogluon/tabular/predictor/predictor.py

@@ -57,7 +57,6 @@ from ..learner import AbstractTabularLearner, DefaultLearner
 from ..trainer.abstract_trainer import AbstractTabularTrainer
 from ..trainer.model_presets.presets import MODEL_TYPES
 from ..version import __version__
-from ._deprecated_methods import TabularPredictorDeprecatedMixin
 
 logger = logging.getLogger(__name__)  # return autogluon root logger
 
@@ -68,7 +67,7 @@ logger = logging.getLogger(__name__)  # return autogluon root logger
 # TODO: consider adding kwarg option for data which has already been preprocessed by feature generator to skip feature generation.
 # TODO: Resolve raw text feature usage in default feature generator
 # TODO: num_bag_sets -> ag_args
-class TabularPredictor(TabularPredictorDeprecatedMixin):
+class TabularPredictor:
     """
     AutoGluon TabularPredictor predicts values in a column of a tabular dataset (classification or regression).
 
@@ -3793,6 +3792,80 @@ class TabularPredictor(TabularPredictorDeprecatedMixin):
         self._assert_is_fit("info")
         return self._learner.get_info(include_model_info=True, include_model_failures=True)
 
+    def model_info(self, model: str) -> dict:
+        """
+        Returns metadata information about the given model.
+        Equivalent output to `predictor.info()["model_info"][model]`
+
+        Parameters
+        ----------
+        model: str
+            The name of the model to get info for.
+
+        Returns
+        -------
+        model_info: dict
+            Model info dictionary
+
+        """
+        return self._trainer.get_model_info(model=model)
+
+    # TODO: Add entire `hyperparameters` dict method for multiple models (including stack ensemble)
+    # TODO: Add unit test
+    def model_hyperparameters(
+        self,
+        model: str,
+        include_ag_args_ensemble: bool = True,
+        output_format: Literal["user", "all"] = "user",
+    ) -> dict:
+        """
+        Returns the hyperparameters of a given model.
+
+        Parameters
+        ----------
+        model: str
+            The name of the model to get hyperparameters for.
+        include_ag_args_ensemble: bool, default True
+            If True, includes the ag_args_ensemble parameters if they exist (for example, when bagging is enabled).
+        output_format: {"user", "all"}, default "user"
+            If "user", returns the same hyperparameters specified by the user (only non-defaults).
+            If "all", returns all hyperparameters used by the model (including default hyperparameters not specified by the user)
+            Regardless of the output_format, they both are functionally equivalent if passed to AutoGluon.
+
+        Returns
+        -------
+        model_hyperparameters: dict
+            Dictionary of model hyperparameters.
+            Equivalent to the model_hyperparameters specified by the user for this model in:
+                `predictor.fit(..., hyperparameters={..., model_key: [..., model_hyperparameters]})`
+
+        """
+        # TODO: Move logic into trainer?
+        info_model = self.model_info(model=model)
+        if output_format == "user":
+            if "bagged_info" in info_model:
+                hyperparameters = info_model["bagged_info"]["child_hyperparameters_user"].copy()
+                if include_ag_args_ensemble and info_model["hyperparameters_user"]:
+                    hyperparameters["ag_args_ensemble"] = info_model["hyperparameters_user"]
+            else:
+                hyperparameters = info_model["hyperparameters_user"]
+        elif output_format == "all":
+            if "bagged_info" in info_model:
+                hyperparameters = info_model["bagged_info"]["child_hyperparameters"].copy()
+                if info_model["bagged_info"]["child_ag_args_fit"]:
+                    hyperparameters["ag_args_fit"] = info_model["bagged_info"]["child_ag_args_fit"]
+                if include_ag_args_ensemble:
+                    bag_hyperparameters = info_model["hyperparameters"].copy()
+                    if info_model["ag_args_fit"]:
+                        bag_hyperparameters["ag_args_fit"] = info_model["ag_args_fit"]
+                    if bag_hyperparameters:
+                        hyperparameters["ag_args_ensemble"] = bag_hyperparameters
+            else:
+                hyperparameters = info_model["hyperparameters"]
+        else:
+            raise ValueError(f"output_format={output_format} is unknown!")
+        return hyperparameters
+
     # TODO: Add data argument
     # TODO: Add option to disable OOF generation of newly fitted models
     # TODO: Move code logic to learner/trainer

{autogluon.tabular-1.2.1b20250218 → autogluon.tabular-1.2.1b20250220}/src/autogluon/tabular/version.py

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

{autogluon.tabular-1.2.1b20250218 → autogluon.tabular-1.2.1b20250220}/src/autogluon.tabular.egg-info/PKG-INFO

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

{autogluon.tabular-1.2.1b20250218 → autogluon.tabular-1.2.1b20250220}/src/autogluon.tabular.egg-info/SOURCES.txt

@@ -151,7 +151,6 @@ src/autogluon/tabular/models/xgboost/hyperparameters/searchspaces.py
 src/autogluon/tabular/models/xt/__init__.py
 src/autogluon/tabular/models/xt/xt_model.py
 src/autogluon/tabular/predictor/__init__.py
-src/autogluon/tabular/predictor/_deprecated_methods.py
 src/autogluon/tabular/predictor/interpretable_predictor.py
 src/autogluon/tabular/predictor/predictor.py
 src/autogluon/tabular/trainer/__init__.py

{autogluon.tabular-1.2.1b20250218 → autogluon.tabular-1.2.1b20250220}/src/autogluon.tabular.egg-info/requires.txt

@@ -3,20 +3,20 @@ scipy<1.16,>=1.5.4
 pandas<2.3.0,>=2.0.0
 scikit-learn<1.5.3,>=1.4.0
 networkx<4,>=3.0
-autogluon.core==1.2.1b20250218
-autogluon.features==1.2.1b20250218
+autogluon.core==1.2.1b20250220
+autogluon.features==1.2.1b20250220
 
 [all]
-huggingface_hub[torch]
-xgboost<2.2,>=1.6
+catboost<1.3,>=1.2
+fastai<2.8,>=2.3.1
+autogluon.core[all]==1.2.1b20250220
 lightgbm<4.6,>=4.0
-spacy<3.8
-autogluon.core[all]==1.2.1b20250218
+huggingface_hub[torch]
 numpy<2.0.0,>=1.25
-einops<0.9,>=0.7
-catboost<1.3,>=1.2
+xgboost<2.2,>=1.6
 torch<2.6,>=2.2
-fastai<2.8,>=2.3.1
+einops<0.9,>=0.7
+spacy<3.8
 
 [catboost]
 numpy<2.0.0,>=1.25
@@ -34,7 +34,7 @@ imodels<1.4.0,>=1.3.10
 lightgbm<4.6,>=4.0
 
 [ray]
-autogluon.core[all]==1.2.1b20250218
+autogluon.core[all]==1.2.1b20250220
 
 [skex]
 scikit-learn-intelex<2025.1,>=2024.0

autogluon.tabular-1.2.1b20250218/src/autogluon/tabular/predictor/_deprecated_methods.py

@@ -1,65 +0,0 @@
-from __future__ import annotations
-
-import numpy as np
-import pandas as pd
-
-from autogluon.common.utils import Deprecated
-
-
-class TabularPredictorDeprecatedMixin:
-    """Contains deprecated methods from TabularPredictor that shouldn't show up in API documentation."""
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="persist")
-    def persist_models(self, *args, **kwargs) -> list[str]:
-        """Deprecated method. Use `persist` instead."""
-        return self.persist(*args, **kwargs)
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="unpersist")
-    def unpersist_models(self, *args, **kwargs) -> list[str]:
-        """Deprecated method. Use `unpersist` instead."""
-        return self.unpersist(*args, **kwargs)
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="model_names")
-    def get_model_names(self, *args, **kwargs) -> list[str]:
-        """Deprecated method. Use `model_names` instead."""
-        return self.model_names(*args, **kwargs)
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="model_best")
-    def get_model_best(self) -> str:
-        """Deprecated method. Use `model_best` instead."""
-        return self.model_best
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="predict_from_proba")
-    def get_pred_from_proba(self, *args, **kwargs) -> pd.Series | np.array:
-        """Deprecated method. Use `predict_from_proba` instead."""
-        return self.predict_from_proba(*args, **kwargs)
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="model_refit_map")
-    def get_model_full_dict(self, *args, **kwargs) -> dict[str, str]:
-        """Deprecated method. Use `model_refit_map` instead."""
-        return self.model_refit_map(*args, **kwargs)
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="predict_proba_oof")
-    def get_oof_pred_proba(self, *args, **kwargs) -> pd.DataFrame | pd.Series:
-        """Deprecated method. Use `predict_proba_oof` instead."""
-        return self.predict_proba_oof(*args, **kwargs)
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="predict_oof")
-    def get_oof_pred(self, *args, **kwargs) -> pd.Series:
-        """Deprecated method. Use `predict_oof` instead."""
-        return self.predict_oof(*args, **kwargs)
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="disk_usage_per_file")
-    def get_size_disk_per_file(self, *args, **kwargs) -> pd.Series:
-        """Deprecated method. Use `disk_usage_per_file` instead."""
-        return self.disk_usage_per_file(*args, **kwargs)
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="disk_usage")
-    def get_size_disk(self) -> int:
-        """Deprecated method. Use `disk_usage` instead."""
-        return self.disk_usage()
-
-    @Deprecated(min_version_to_warn="0.8.3", min_version_to_error="1.2", version_to_remove="1.2", new="model_names(persisted=True)")
-    def get_model_names_persisted(self) -> list[str]:
-        """Deprecated method. Use `model_names(persisted=True)` instead."""
-        return self.model_names(persisted=True)