autogluon.common 1.2.1b20250409__py3-none-any.whl → 1.2.1b20250410__py3-none-any.whl

autogluon/common/utils/simulation_utils.py

@@ -1,6 +1,7 @@
 from collections import defaultdict
 from typing import Any, Dict, Tuple
 
+import numpy as np
 import pandas as pd
 
 
@@ -66,8 +67,45 @@ def convert_simulation_artifacts_to_tabular_predictions_dict(
                     "label",
                 ]:
                     aggregated_ground_truth[task_name][fold][k] = zeroshot_metadata[k]
-            for k in ["pred_proba_dict_val", "pred_proba_dict_test"]:
+            ordered_class_labels_transformed = aggregated_ground_truth[task_name][fold][
+                "ordered_class_labels_transformed"
+            ]
+            pred_proba_tuples = [
+                ("pred_proba_dict_val", "y_val", "y_val_idx"),
+                ("pred_proba_dict_test", "y_test", "y_test_idx"),
+            ]
+            for k, ground_truth_key, ground_truth_idx_key in pred_proba_tuples:
+                ground_truth_idx = zeroshot_metadata.get(ground_truth_idx_key, None)
+
+                if ground_truth_idx is not None:
+                    ground_truth_np = zeroshot_metadata[ground_truth_key]
+                    assert isinstance(ground_truth_np, np.ndarray)
+                    assert isinstance(ground_truth_idx, np.ndarray)
+                    # memory opt variant in np.ndarray format, convert to pandas
+                    if len(ground_truth_np.shape) == 1:
+                        ground_truth_pd = pd.Series(
+                            data=ground_truth_np, index=ground_truth_idx, name=zeroshot_metadata["label"]
+                        )
+                    else:
+                        ground_truth_pd = pd.DataFrame(
+                            data=ground_truth_np, index=ground_truth_idx, columns=ordered_class_labels_transformed
+                        )
+                    aggregated_ground_truth[task_name][fold][ground_truth_key] = ground_truth_pd
+                ground_truth = aggregated_ground_truth[task_name][fold][ground_truth_key]
+                assert isinstance(ground_truth, (pd.Series, pd.DataFrame))
                 for m, pred_proba in zeroshot_metadata[k].items():
+                    if isinstance(pred_proba, np.ndarray):
+                        if len(pred_proba.shape) == 1:
+                            if ordered_class_labels_transformed is not None:
+                                assert len(ordered_class_labels_transformed) == 2
+                                series_name = ordered_class_labels_transformed[1]
+                            else:
+                                series_name = ground_truth.name
+                            pred_proba = pd.Series(data=pred_proba, index=ground_truth.index, name=series_name)
+                        else:
+                            pred_proba = pd.DataFrame(
+                                data=pred_proba, index=ground_truth.index, columns=ordered_class_labels_transformed
+                            )
                     if aggregated_ground_truth[task_name][fold]["problem_type"] == "binary":
                         if isinstance(pred_proba, pd.DataFrame):
                             assert len(pred_proba.columns) == 2

autogluon/common/version.py

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

{autogluon.common-1.2.1b20250409.dist-info → autogluon.common-1.2.1b20250410.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.common
-Version: 1.2.1b20250409
+Version: 1.2.1b20250410
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -42,10 +42,10 @@ Requires-Dist: boto3<2,>=1.10
 Requires-Dist: psutil<7.0.0,>=5.7.3
 Requires-Dist: tqdm<5,>=4.38
 Provides-Extra: tests
-Requires-Dist: types-requests; extra == "tests"
-Requires-Dist: pytest-mypy; extra == "tests"
 Requires-Dist: types-setuptools; extra == "tests"
 Requires-Dist: pytest; extra == "tests"
+Requires-Dist: pytest-mypy; extra == "tests"
+Requires-Dist: types-requests; extra == "tests"
 
 
 

{autogluon.common-1.2.1b20250409.dist-info → autogluon.common-1.2.1b20250410.dist-info}/RECORD

@@ -1,8 +1,8 @@
-autogluon.common-1.2.1b20250409-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.common-1.2.1b20250410-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/common/__init__.py,sha256=Q0tQ4UOtUEhPpj83k_xzQ9bmtoTSvfg_3WgPi1QopJk,352
 autogluon/common/dataset.py,sha256=slfU8CxGFNJ86_2wKkT_zPMPr-iPB-sz7klz0YqJECo,1580
 autogluon/common/space.py,sha256=oRSG-uFmsLACT42mT2vRp3NSuV8MaQz6X4sYwcZruxs,5457
-autogluon/common/version.py,sha256=A9BqrXhLrOJhH5qPbhMof-jLGxRS5nRGGDYPGWeziJE,91
+autogluon/common/version.py,sha256=tJeST0EZCa1sWbuBSG5OURtJwcs6LfjpwjLa8Ns8AI4,91
 autogluon/common/features/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/common/features/feature_metadata.py,sha256=GDvZXyaNkDeuNv8omO4K7tkCPmMdpJzD5rH_hKtZD9E,25783
 autogluon/common/features/infer_types.py,sha256=MkfVDoY__0Rr__zsDSKszmvYMq5RQRmbNboTG_tm64o,6492
@@ -39,15 +39,15 @@ autogluon/common/utils/pandas_utils.py,sha256=sirUUQkKZsmMwmXWEUZJ1n3ZYFQWqwz4iu
 autogluon/common/utils/path_converter.py,sha256=OO0Jr5vhcqrEvZrNqem5tmgC6FZA3_O8sPylQUKUgKs,2441
 autogluon/common/utils/resource_utils.py,sha256=CmU8OsaXoreja43gMGRZCg8BiKQdHvDxWW0y7auZE2Q,8456
 autogluon/common/utils/s3_utils.py,sha256=Tj2VyUGUc7Vrx5KkCGKapvFQyaVvWpJNpjssfXg2Sl4,17142
-autogluon/common/utils/simulation_utils.py,sha256=iEZcRWzP8xLQ8racIfOGC5V_A0W_xNgxPkGuBfX6hh8,3573
+autogluon/common/utils/simulation_utils.py,sha256=hzDv0W7uPloadyXRNiAwWGs8IYhdXqP724gkYRo_k6I,5880
 autogluon/common/utils/system_info.py,sha256=9BkFcdq1co0EMnZkk_AYdhxSEP3RqcU6uIp-iYx9bdc,4479
 autogluon/common/utils/try_import.py,sha256=dGNpTev0egEg-NsdPWYRO4dbjIvMwU0AN7Ky5miv0MI,6787
 autogluon/common/utils/utils.py,sha256=H0gjBbiFBPuhnlHpirN3nU7WehoTZCv-231tb1oIvMI,7971
-autogluon.common-1.2.1b20250409.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.common-1.2.1b20250409.dist-info/METADATA,sha256=_6dOs4jLfLgdeJlVbQdhhy9GwCyp2Q-RZcVZ_ldhc8s,11658
-autogluon.common-1.2.1b20250409.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.common-1.2.1b20250409.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.common-1.2.1b20250409.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.common-1.2.1b20250409.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.common-1.2.1b20250409.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.common-1.2.1b20250409.dist-info/RECORD,,
+autogluon.common-1.2.1b20250410.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.common-1.2.1b20250410.dist-info/METADATA,sha256=RI1JnwN_cl-Anxd7LLnZbC36erX4Kc8hWQwIrqC9FIQ,11658
+autogluon.common-1.2.1b20250410.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.common-1.2.1b20250410.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.common-1.2.1b20250410.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.common-1.2.1b20250410.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.common-1.2.1b20250410.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.common-1.2.1b20250410.dist-info/RECORD,,