autogluon.tabular 1.2.1b20250113__py3-none-any.whl → 1.2.1b20250114__py3-none-any.whl

autogluon/tabular/predictor/predictor.py

@@ -1388,10 +1388,19 @@ class TabularPredictor(TabularPredictorDeprecatedMixin):
             )
         else:
             # Holdout is false, use (repeated) cross-validation
-            is_stratified = self.problem_type in [REGRESSION, QUANTILE, SOFTCLASS]
+            is_stratified = self.problem_type in [BINARY, MULTICLASS]
+            is_binned = self.problem_type in [REGRESSION, QUANTILE]
             self._learner._validate_groups(X=X, X_val=X_val)  # Validate splits before splitting
-            splits = CVSplitter(n_splits=n_folds, n_repeats=n_repeats, groups=self._learner.groups, stratified=is_stratified, random_state=42).split(
-                X=X.drop(self.label, axis=1), y=X[self.label]
+            splits = CVSplitter(
+                n_splits=n_folds,
+                n_repeats=n_repeats,
+                groups=self._learner.groups,
+                stratify=is_stratified,
+                bin=is_binned,
+                random_state=42,
+            ).split(
+                X=X.drop(self.label, axis=1),
+                y=X[self.label]
             )
             n_splits = len(splits)
             logger.info(

autogluon/tabular/version.py

@@ -1,3 +1,3 @@
 """This is the autogluon version file."""
-__version__ = '1.2.1b20250113'
+__version__ = '1.2.1b20250114'
 __lite__ = False

{autogluon.tabular-1.2.1b20250113.dist-info → autogluon.tabular-1.2.1b20250114.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.2.1b20250113
+Version: 1.2.1b20250114
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -39,19 +39,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.1b20250113
-Requires-Dist: autogluon.features==1.2.1b20250113
+Requires-Dist: autogluon.core==1.2.1b20250114
+Requires-Dist: autogluon.features==1.2.1b20250114
 Provides-Extra: all
-Requires-Dist: spacy<3.8; extra == "all"
+Requires-Dist: numpy<2.0.0,>=1.25; extra == "all"
+Requires-Dist: fastai<2.8,>=2.3.1; extra == "all"
 Requires-Dist: lightgbm<4.6,>=4.0; extra == "all"
 Requires-Dist: torch<2.6,>=2.2; extra == "all"
-Requires-Dist: autogluon.core[all]==1.2.1b20250113; extra == "all"
-Requires-Dist: fastai<2.8,>=2.3.1; extra == "all"
+Requires-Dist: spacy<3.8; extra == "all"
 Requires-Dist: xgboost<2.2,>=1.6; extra == "all"
-Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: autogluon.core[all]==1.2.1b20250114; extra == "all"
 Requires-Dist: einops<0.9,>=0.7; extra == "all"
+Requires-Dist: catboost<1.3,>=1.2; extra == "all"
 Requires-Dist: huggingface-hub[torch]; extra == "all"
-Requires-Dist: numpy<2.0.0,>=1.25; extra == "all"
 Provides-Extra: catboost
 Requires-Dist: numpy<2.0.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
@@ -64,7 +64,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.1b20250113; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.2.1b20250114; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.1,>=2024.0; extra == "skex"
 Provides-Extra: skl2onnx
@@ -74,13 +74,13 @@ 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: tabpfn
-Requires-Dist: tabpfn<0.2,>=0.1; extra == "tabpfn"
+Requires-Dist: tabpfn<0.2,>=0.1.11; extra == "tabpfn"
 Provides-Extra: tabpfnmix
 Requires-Dist: torch<2.6,>=2.2; extra == "tabpfnmix"
 Requires-Dist: huggingface-hub[torch]; extra == "tabpfnmix"
 Requires-Dist: einops<0.9,>=0.7; extra == "tabpfnmix"
 Provides-Extra: tests
-Requires-Dist: tabpfn<0.2,>=0.1; extra == "tests"
+Requires-Dist: tabpfn<0.2,>=0.1.11; extra == "tests"
 Requires-Dist: torch<2.6,>=2.2; extra == "tests"
 Requires-Dist: huggingface-hub[torch]; extra == "tests"
 Requires-Dist: einops<0.9,>=0.7; extra == "tests"

{autogluon.tabular-1.2.1b20250113.dist-info → autogluon.tabular-1.2.1b20250114.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.2.1b20250113-py3.8-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.2.1b20250114-py3.8-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=xp_3RhUZoc8du55mJV_5b3-ffuLY-6zGvseBIwCs1Z0,90
+autogluon/tabular/version.py,sha256=Um-j8InMDDRIJ5WoL607BMYYvkzsVWU8cxXJ2aXfjbQ,90
 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
@@ -144,7 +144,7 @@ autogluon/tabular/models/xt/xt_model.py,sha256=gimCfMeTmhAJTy4ekCI_dbK1FluJ7EQMo
 autogluon/tabular/predictor/__init__.py,sha256=zCMgjxQlWpDWnr1l1xjBCiK3rWC3N3RoD8UXBnazT74,107
 autogluon/tabular/predictor/_deprecated_methods.py,sha256=N0QqI9wElXdSeQ9idnnJpoH2xkmv51j_vG7kDrP5QxU,3507
 autogluon/tabular/predictor/interpretable_predictor.py,sha256=5UeKgnMFsfY65tiO3kxfHBPr03lyswLrgdtjPhI0Y7Q,6934
-autogluon/tabular/predictor/predictor.py,sha256=3MvREMsmZOds_wPrRbbLuVRoIBPzk88-ck7KpfQ-xlQ,352250
+autogluon/tabular/predictor/predictor.py,sha256=-0d5g_ADcSCwJKS6uJVif9iJD3Sq6FgcVFPrPRZ-l1Q,352445
 autogluon/tabular/trainer/__init__.py,sha256=PW_PGL-tWoQzx3ES2S53bQEZOtsRWTYiM9QdOqsk0dI,38
 autogluon/tabular/trainer/auto_trainer.py,sha256=ieu7Zofl93tnQC9sIcsvS9pJS__KVIsE7Nn-EIWRvT0,8718
 autogluon/tabular/trainer/model_presets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -152,11 +152,11 @@ autogluon/tabular/trainer/model_presets/presets.py,sha256=1E-Z1FxUpyydaoEdxcTCg7
 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.1b20250113.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.2.1b20250113.dist-info/METADATA,sha256=JJQA0WoFX7ILBBGWa67GZljZAyoDymm11xWKAxVC6iw,14309
-autogluon.tabular-1.2.1b20250113.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.2.1b20250113.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.2.1b20250113.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250113.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.2.1b20250113.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.2.1b20250113.dist-info/RECORD,,
+autogluon.tabular-1.2.1b20250114.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.2.1b20250114.dist-info/METADATA,sha256=yz9jxr4weyLJLpGkDAqM-ZSWeAnSCC_xU8w1zmbHxpE,14315
+autogluon.tabular-1.2.1b20250114.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.2.1b20250114.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.2.1b20250114.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250114.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.2.1b20250114.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.2.1b20250114.dist-info/RECORD,,