autogluon.tabular 1.3.2b20250614__py3-none-any.whl → 1.3.2b20250616__py3-none-any.whl

autogluon/tabular/models/_utils/rapids_utils.py

@@ -10,7 +10,7 @@ class RapidsModelMixin:
     @classmethod
     def _get_default_ag_args_ensemble(cls, **kwargs) -> dict:
         default_ag_args_ensemble = super()._get_default_ag_args_ensemble(**kwargs)
-        extra_ag_args_ensemble = {"use_child_oof": False}
+        extra_ag_args_ensemble = {"use_child_oof": False, "fold_fitting_strategy": "sequential_local"}
         default_ag_args_ensemble.update(extra_ag_args_ensemble)
         return default_ag_args_ensemble
 

autogluon/tabular/models/lr/lr_rapids_model.py

@@ -1,7 +1,5 @@
 import logging
 
-import numpy as np
-
 from autogluon.common.utils.try_import import try_import_rapids_cuml
 from autogluon.core.constants import REGRESSION
 
@@ -51,10 +49,52 @@ class LinearRapidsModel(RapidsModelMixin, LinearModel):
 
     def _preprocess(self, X, **kwargs):
         X = super()._preprocess(X=X, **kwargs)
-        if not isinstance(X, np.ndarray):
+        if hasattr(X, 'toarray'):  # Check if it's a sparse matrix
             X = X.toarray()
         return X
 
     def _fit(self, X, y, **kwargs):
-        kwargs.pop("sample_weight", None)  # sample_weight is not supported
-        super()._fit(X=X, y=y, **kwargs)
+        """
+        Custom fit method for RAPIDS cuML models that handles parameter compatibility
+        and bypasses sklearn-specific incremental training approach.
+        """
+        # Preprocess data
+        X = self.preprocess(X, is_train=True)
+        if self.problem_type == 'binary':
+            y = y.astype(int).values
+
+        # Create cuML model with filtered parameters
+        model_cls = self._get_model_type()
+
+        # Comprehensive parameter filtering for cuML compatibility
+        cuml_incompatible_params = {
+            # AutoGluon-specific preprocessing parameters
+            'vectorizer_dict_size', 'proc.ngram_range', 'proc.skew_threshold',
+            'proc.impute_strategy', 'handle_text',
+            # sklearn-specific parameters not supported by cuML
+            'n_jobs', 'warm_start', 'multi_class', 'dual', 'intercept_scaling',
+            'class_weight', 'random_state', 'verbose',
+            # Parameters that need conversion or special handling
+            'penalty', 'C'
+        }
+
+        # Filter out incompatible parameters
+        filtered_params = {k: v for k, v in self.params.items()
+                          if k not in cuml_incompatible_params}
+
+        # Handle parameter conversions for cuML
+        if self.problem_type == REGRESSION:
+            # Convert sklearn's C parameter to cuML's alpha
+            if 'C' in self.params:
+                filtered_params['alpha'] = 1.0 / self.params['C']
+        else:
+            # For classification, keep C parameter
+            if 'C' in self.params:
+                filtered_params['C'] = self.params['C']
+
+        # Create and fit cuML model - let cuML handle its own error messages
+        self.model = model_cls(**filtered_params)
+        self.model.fit(X, y)
+
+        # Add missing sklearn-compatible attributes for AutoGluon compatibility
+        self.model.n_iter_ = None  # cuML doesn't track iterations like sklearn

autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.3.2b20250614"
+__version__ = "1.3.2b20250616"
 __lite__ = False

{autogluon.tabular-1.3.2b20250614.dist-info → autogluon.tabular-1.3.2b20250616.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: autogluon.tabular
-Version: 1.3.2b20250614
+Version: 1.3.2b20250616
 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.7.0,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.3.2b20250614
-Requires-Dist: autogluon.features==1.3.2b20250614
+Requires-Dist: autogluon.core==1.3.2b20250616
+Requires-Dist: autogluon.features==1.3.2b20250616
 Provides-Extra: all
-Requires-Dist: huggingface-hub[torch]; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
-Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
-Requires-Dist: einops<0.9,>=0.7; extra == "all"
 Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "all"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
 Requires-Dist: torch<2.7,>=2.2; extra == "all"
-Requires-Dist: autogluon.core[all]==1.3.2b20250614; extra == "all"
-Requires-Dist: spacy<3.9; extra == "all"
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: einops<0.9,>=0.7; extra == "all"
+Requires-Dist: autogluon.core[all]==1.3.2b20250616; extra == "all"
+Requires-Dist: xgboost<3.1,>=2.0; extra == "all"
+Requires-Dist: huggingface-hub[torch]; extra == "all"
+Requires-Dist: spacy<3.9; extra == "all"
 Provides-Extra: catboost
 Requires-Dist: numpy<2.3.0,>=1.25; extra == "catboost"
 Requires-Dist: catboost<1.3,>=1.2; extra == "catboost"
@@ -66,7 +66,7 @@ Requires-Dist: imodels<2.1.0,>=1.3.10; extra == "imodels"
 Provides-Extra: lightgbm
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "lightgbm"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.3.2b20250614; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.3.2b20250616; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.5,>=2024.0; extra == "skex"
 Provides-Extra: skl2onnx

{autogluon.tabular-1.3.2b20250614.dist-info → autogluon.tabular-1.3.2b20250616.dist-info}/RECORD

@@ -1,6 +1,6 @@
-autogluon.tabular-1.3.2b20250614-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.3.2b20250616-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=9bO1qnzElFr3y5XhhPTxmVf49lJgK0hlIavjJaLW45M,91
+autogluon/tabular/version.py,sha256=62vzXJ8PVHkbeXOmkBmEiscHahqhaxUjLVRKzCtWNBU,91
 autogluon/tabular/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/configs/config_helper.py,sha256=JsdVGmpcYL88GPKBznPtqJ1sGaByOSvLn7KWU-HyVoQ,21085
 autogluon/tabular/configs/feature_generator_presets.py,sha256=EV5Ym8VW15q92MwOUpTi7wZFS2QooM51fLg3RdUsn-M,1223
@@ -18,7 +18,7 @@ autogluon/tabular/learner/abstract_learner.py,sha256=0kf0huvg0nphe-lrdKtNTzdIFr1
 autogluon/tabular/learner/default_learner.py,sha256=hjdKbcFtIQxQ3-k1LiGOo-w5sLxIIQAyFLs3-R35aw0,24781
 autogluon/tabular/models/__init__.py,sha256=fZDKUKiD9hDzEyFXXbt7_b4yADK9peREdP8QoukWukQ,1036
 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/rapids_utils.py,sha256=9A2Y10Owva6zhcLkBVQ_T4tOAMDp1idSMzDWhl_QyBI,1083
 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=GvrMBC8Z-zobalmSzX1iDHTYMmQ4Jp5hINJa_fSm-j8,11322
@@ -63,7 +63,7 @@ autogluon/tabular/models/lgb/hyperparameters/searchspaces.py,sha256=tvNNR7niWz_B
 autogluon/tabular/models/lr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/lr/lr_model.py,sha256=wTrrTWVwmlAyx4RAxhfXHbkvZTAVIvAiTadpEChGEzc,15599
 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/lr_rapids_model.py,sha256=XIB1KCPPfBZMxTRC3Wc1Dsl5NTMQSM_m8Uc2igyTLX8,3939
 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
@@ -146,11 +146,11 @@ autogluon/tabular/trainer/model_presets/presets.py,sha256=3gM_QFpG_BaVFIf8T0nCd-
 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.3.2b20250614.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.3.2b20250614.dist-info/METADATA,sha256=at6dJWU7j8NcRSdG2NIcu4_YCn79CIj_W9sqAbCZFFI,14069
-autogluon.tabular-1.3.2b20250614.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.3.2b20250614.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.3.2b20250614.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250614.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250614.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.3.2b20250614.dist-info/RECORD,,
+autogluon.tabular-1.3.2b20250616.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon.tabular-1.3.2b20250616.dist-info/METADATA,sha256=8D5eieiD9qeVspY9rX76QKhIAvtYfb6k-UsZdPZD8WA,14069
+autogluon.tabular-1.3.2b20250616.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon.tabular-1.3.2b20250616.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+autogluon.tabular-1.3.2b20250616.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250616.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon.tabular-1.3.2b20250616.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon.tabular-1.3.2b20250616.dist-info/RECORD,,