autogluon.core 1.2.1b20250218__py3-none-any.whl → 1.2.1b20250220__py3-none-any.whl

autogluon/core/models/abstract/abstract_model.py

@@ -1632,9 +1632,7 @@ class AbstractModel:
         path = self.path_root
         problem_type = self.problem_type
         eval_metric = self.eval_metric
-        hyperparameters = self._user_params.copy()
-        if self._user_params_aux:
-            hyperparameters[AG_ARGS_FIT] = self._user_params_aux.copy()
+        hyperparameters = self.get_hyperparameters_init()
 
         args = dict(
             path=path,
@@ -1646,6 +1644,20 @@ class AbstractModel:
 
         return args
 
+    def get_hyperparameters_init(self) -> dict:
+        """
+
+        Returns
+        -------
+        hyperparameters: dict
+            The dictionary of user specified hyperparameters for the model.
+
+        """
+        hyperparameters = self._user_params.copy()
+        if self._user_params_aux:
+            hyperparameters[AG_ARGS_FIT] = self._user_params_aux.copy()
+        return hyperparameters
+
     def convert_to_template(self):
         """
         After calling this function, returned model should be able to be fit as if it was new, as well as deep-copied.
@@ -2227,7 +2239,7 @@ class AbstractModel:
         # TODO: Report errors?
         shutil.rmtree(path=model_path, ignore_errors=True)
 
-    def get_info(self) -> dict:
+    def get_info(self, include_feature_metadata: bool = True) -> dict:
         """
         Returns a dictionary of numerous fields describing the model.
         """
@@ -2243,6 +2255,7 @@ class AbstractModel:
             "predict_time": self.predict_time,
             "val_score": self.val_score,
             "hyperparameters": self.params,
+            "hyperparameters_user": self.get_hyperparameters_init(),
             "hyperparameters_fit": self.params_trained,  # TODO: Explain in docs that this is for hyperparameters that differ in final model from original hyperparameters, such as epochs (from early stopping)
             "hyperparameters_nondefault": self.nondefault_params,
             AG_ARGS_FIT: self.get_params_aux_info(),
@@ -2260,6 +2273,8 @@ class AbstractModel:
         }
         if self._is_fit_metadata_registered:
             info.update(self._fit_metadata)
+        if not include_feature_metadata:
+            info.pop("feature_metadata")
         return info
 
     def get_params_aux_info(self) -> dict:

autogluon/core/models/ensemble/bagged_ensemble_model.py

@@ -1192,6 +1192,28 @@ class BaggedEnsembleModel(AbstractModel):
         init_args.pop("problem_type")
         return init_args
 
+    def get_hyperparameters_init_child(self, include_ag_args_ensemble: bool = False, child_model: AbstractModel = None) -> dict:
+        """
+
+        Returns
+        -------
+        hyperparameters: dict
+            The dictionary of user specified hyperparameters for the model.
+
+        """
+        if child_model is None:
+            if self.n_children > 0:
+                child_model = self.load_child(self.models[0])
+            else:
+                child_model = self._get_model_base()
+        hyperparameters_child = child_model.get_hyperparameters_init()
+        if include_ag_args_ensemble:
+            hyperparameters_self = self.get_hyperparameters_init()
+            if hyperparameters_self:
+                hyperparameters_child["ag_args_ensemble"] = hyperparameters_self
+
+        return hyperparameters_child
+
     def convert_to_template_child(self):
         return self._get_model_base().convert_to_template()
 
@@ -1409,9 +1431,9 @@ class BaggedEnsembleModel(AbstractModel):
                 model_names.append(model.name)
         return model_names
 
-    def get_info(self):
-        info = super().get_info()
-        children_info = self._get_child_info()
+    def get_info(self, include_feature_metadata: bool = True):
+        info = super().get_info(include_feature_metadata=include_feature_metadata)
+        children_info = self._get_child_info(include_feature_metadata=include_feature_metadata)
         child_memory_sizes = [child["memory_size"] for child in children_info.values()]
         sum_memory_size_child = sum(child_memory_sizes)
         if child_memory_sizes:
@@ -1432,6 +1454,7 @@ class BaggedEnsembleModel(AbstractModel):
             child_model = self._get_model_base()
         child_hyperparameters = child_model.params
         child_ag_args_fit = child_model.params_aux
+        child_hyperparameters_user = self.get_hyperparameters_init_child(include_ag_args_ensemble=False, child_model=child_model)
 
         bagged_info = dict(
             child_model_type=self._child_type.__name__,
@@ -1448,6 +1471,7 @@ class BaggedEnsembleModel(AbstractModel):
             max_memory_size=max_memory_size,  # Memory used when all children are loaded into memory at once.
             min_memory_size=min_memory_size,  # Memory used when only the largest child is loaded into memory.
             child_hyperparameters=child_hyperparameters,
+            child_hyperparameters_user=child_hyperparameters_user,
             child_hyperparameters_fit=self._get_compressed_params_trained(),
             child_ag_args_fit=child_ag_args_fit,
         )
@@ -1480,14 +1504,16 @@ class BaggedEnsembleModel(AbstractModel):
         # memory is checked downstream on the child model
         pass
 
-    def _get_child_info(self):
+    def _get_child_info(self, include_feature_metadata: bool = True):
         child_info_dict = dict()
         for model in self.models:
             if isinstance(model, str):
                 child_path = self.create_contexts(os.path.join(self.path, model))
                 child_info_dict[model] = self._child_type.load_info(child_path)
+                if not include_feature_metadata:
+                    child_info_dict[model].pop("feature_metadata", None)
             else:
-                child_info_dict[model.name] = model.get_info()
+                child_info_dict[model.name] = model.get_info(include_feature_metadata=include_feature_metadata)
         return child_info_dict
 
     def _construct_empty_oof(self, X, y):

autogluon/core/models/ensemble/stacker_ensemble_model.py

@@ -309,8 +309,8 @@ class StackerEnsembleModel(BaggedEnsembleModel):
             model = model_type.load(model_path)
         return model
 
-    def get_info(self):
-        info = super().get_info()
+    def get_info(self, **kwargs):
+        info = super().get_info(**kwargs)
         stacker_info = dict(
             num_base_models=len(self.base_model_names),
             base_model_names=self.base_model_names,

autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py

@@ -111,8 +111,8 @@ class GreedyWeightedEnsembleModel(AbstractModel):
         model_weight_dict = {self.base_model_names[i]: self.weights_[i] for i in range(num_models)}
         return model_weight_dict
 
-    def get_info(self):
-        info = super().get_info()
+    def get_info(self, **kwargs):
+        info = super().get_info(**kwargs)
         info["model_weights"] = self._get_model_weights()
         return info
 

autogluon/core/version.py

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

{autogluon.core-1.2.1b20250218.dist-info → autogluon.core-1.2.1b20250220.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.core
-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
@@ -45,24 +45,24 @@ Requires-Dist: tqdm<5,>=4.38
 Requires-Dist: requests
 Requires-Dist: matplotlib<3.11,>=3.7.0
 Requires-Dist: boto3<2,>=1.10
-Requires-Dist: autogluon.common==1.2.1b20250218
+Requires-Dist: autogluon.common==1.2.1b20250220
 Provides-Extra: all
+Requires-Dist: ray[default,tune]<2.43,>=2.10.0; extra == "all"
+Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "all"
 Requires-Dist: pyarrow>=15.0.0; extra == "all"
-Requires-Dist: ray[default]<2.41,>=2.10.0; extra == "all"
 Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "all"
-Requires-Dist: ray[default,tune]<2.41,>=2.10.0; extra == "all"
 Provides-Extra: ray
-Requires-Dist: ray[default]<2.41,>=2.10.0; extra == "ray"
+Requires-Dist: ray[default]<2.43,>=2.10.0; extra == "ray"
 Provides-Extra: raytune
 Requires-Dist: pyarrow>=15.0.0; extra == "raytune"
-Requires-Dist: ray[default,tune]<2.41,>=2.10.0; extra == "raytune"
+Requires-Dist: ray[default,tune]<2.43,>=2.10.0; extra == "raytune"
 Requires-Dist: hyperopt<0.2.8,>=0.2.7; extra == "raytune"
 Provides-Extra: tests
+Requires-Dist: pytest-mypy; extra == "tests"
+Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: flake8; extra == "tests"
 Requires-Dist: types-requests; extra == "tests"
-Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: pytest; extra == "tests"
-Requires-Dist: pytest-mypy; extra == "tests"
 
 
 

{autogluon.core-1.2.1b20250218.dist-info → autogluon.core-1.2.1b20250220.dist-info}/RECORD

@@ -1,9 +1,9 @@
-autogluon.core-1.2.1b20250218-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.core-1.2.1b20250220-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/core/__init__.py,sha256=8KfvvHzXX3a4q6z43Dw1yE7VtbAoiSMaglVpKDy6Xeg,245
 autogluon/core/_setup_utils.py,sha256=ikxn4zc3PNyjJJT5SsgL0dvP-6Rbq6_dItGMiZNINv4,6958
 autogluon/core/constants.py,sha256=nEVLdSFJ-5O-tz3jUD3qPX65RMp7g8qOR38XlurbP4Y,3403
 autogluon/core/problem_type.py,sha256=XJmMgeNBgS7u43pDK-spTivatPyh_INOXveEXwQt-Rw,2993
-autogluon/core/version.py,sha256=7H9aiBFc3yZ4VAfYTXCxWQ-TNBEjD1nB5tqyO9XpoOs,91
+autogluon/core/version.py,sha256=5_5isKRJ9Kd7emAdEPGzl72mtp4GDkqLCtc4C6IuV6k,91
 autogluon/core/augmentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/augmentation/distill_utils.py,sha256=JBlp2WOMNKoJv8aKVwJVRQSalSk8jx36HM7-k_VvkhY,9404
 autogluon/core/calibrate/__init__.py,sha256=eU6qLj7DKUhaz2HHNHDrfroRaLM-mhuSncK_v1UP4F8,62
@@ -40,21 +40,21 @@ autogluon/core/models/__init__.py,sha256=dg3onYq5wW3-sfdNurnSIGpX0rpEjG_abgzyfwD
 autogluon/core/models/_utils.py,sha256=qswE9n1ge1AJSExgstEbrZiMFmMRa4Mf5Sz8D9-XU6c,2091
 autogluon/core/models/abstract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/abstract/_tags.py,sha256=Qr_3an0ZMig24S3OwISa-nTFfWHQe3pwPTiXq4zlEec,3409
-autogluon/core/models/abstract/abstract_model.py,sha256=SPknoRqTiRCZfTvoFrf2V3BW4h5MVbrl-2HvSHfCjOM,120532
+autogluon/core/models/abstract/abstract_model.py,sha256=eEGAfyQAklkUKHasSRZYrRPKBG2WEYu1ygzYk1eiiK8,121024
 autogluon/core/models/abstract/abstract_nn_model.py,sha256=IId0ivO8uVvmpnK9OiM2CtPVrP1ewOaQQKtQUDtK7_k,4818
 autogluon/core/models/abstract/model_trial.py,sha256=PKEo1jfLSBCOLM42QE5VBD1u41MaVMRk31zhNhLiqTw,5035
 autogluon/core/models/dummy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/dummy/_dummy_quantile_regressor.py,sha256=i-ZW2flJ60jsMfMK24IP39Xwc55-UlBDvHmqanIf29Q,664
 autogluon/core/models/dummy/dummy_model.py,sha256=at2FZSM2_LuAQ78E2YrRCRt3UaKMyyOnc6p2rtZgA2w,1414
 autogluon/core/models/ensemble/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/core/models/ensemble/bagged_ensemble_model.py,sha256=8GijtiLwFXugKWWIKY4ss_Xa_EGdASWtvJ2JUJu2avs,75887
+autogluon/core/models/ensemble/bagged_ensemble_model.py,sha256=kQbJhq_UzUoOHsTX00WTEXu1Ogdme4bdLiiWhIMNUfU,77257
 autogluon/core/models/ensemble/fold_fitting_strategy.py,sha256=01vzNVvE4FIFgD6YqbhK63XoUlSztnVFsrDdsoqm75U,47021
 autogluon/core/models/ensemble/ray_parallel_fold_fitting_strategy.py,sha256=8RASa-eV6n9kUgbqQHNt7k4IrvuB9NdrunIMLYOLwgA,2068
-autogluon/core/models/ensemble/stacker_ensemble_model.py,sha256=DuDXgozvG9JYYkRvGACA7EXDAtj3Tz_uAjXTfxu5tFg,18041
+autogluon/core/models/ensemble/stacker_ensemble_model.py,sha256=oGgn0oxu1jzygcCHKBvlMZLbKrDAIgcrOMXFbSTNTVM,18059
 autogluon/core/models/ensemble/weighted_ensemble_model.py,sha256=hELT3L7tDRYuN7Ghm_9zNH4zzG_g8vk0hINofu_ZqSE,4318
 autogluon/core/models/greedy_ensemble/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/models/greedy_ensemble/ensemble_selection.py,sha256=Rt5jkqIORaCFLiQlh-LVWKP5-eB6gxEnVNmKTB3RP58,10548
-autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py,sha256=GaEfdK4rhfJ06h-LYZCqYshEdFJcCzPQgtYPKw-Jjc8,7023
+autogluon/core/models/greedy_ensemble/greedy_weighted_ensemble_model.py,sha256=VgwuHhzmARE7iWZuCTg0cX2DQnEgLvhlLyWOBYDnpNE,7041
 autogluon/core/pseudolabeling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/core/pseudolabeling/pseudolabeling.py,sha256=vvOrY02Wvba3UGujYH3xqgI6yf2jRegzLiNfKrxSais,12138
 autogluon/core/ray/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -89,11 +89,11 @@ autogluon/core/utils/utils.py,sha256=FMa9kIUAxA3IIBbATmBnNEVObSAivehZ2_zCy3PRR-c
 autogluon/core/utils/version_utils.py,sha256=5-r8hLRKTaZbj5qo2uzE_2E4casH49Ye3WyeHlgHuz4,3252
 autogluon/core/utils/loaders/__init__.py,sha256=W5FAdQvpDcn_uisqJrlSAObWVta-YjJLKGN3NCbEgIo,109
 autogluon/core/utils/savers/__init__.py,sha256=bGWciSxAkj6u06vOC4pTvr22f_1ey0glgvmjCMEOm78,89
-autogluon.core-1.2.1b20250218.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.core-1.2.1b20250218.dist-info/METADATA,sha256=y94iLSyt3ozEWFUl_EpRXiYTiXu8E2UG2_aZbeai58c,12399
-autogluon.core-1.2.1b20250218.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.core-1.2.1b20250218.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.core-1.2.1b20250218.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250218.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.core-1.2.1b20250218.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.core-1.2.1b20250218.dist-info/RECORD,,
+autogluon.core-1.2.1b20250220.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.core-1.2.1b20250220.dist-info/METADATA,sha256=HgWuZIrv6P0SFTOBMIh6U_ialUBNYdhSnK-Q5eeTits,12399
+autogluon.core-1.2.1b20250220.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.core-1.2.1b20250220.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.core-1.2.1b20250220.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250220.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.core-1.2.1b20250220.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.core-1.2.1b20250220.dist-info/RECORD,,