autogluon.tabular 1.5.0b20251230__tar.gz → 1.5.1b20260108__tar.gz

{autogluon_tabular-1.5.0b20251230/src/autogluon.tabular.egg-info → autogluon_tabular-1.5.1b20260108}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: autogluon.tabular
-Version: 1.5.0b20251230
+Version: 1.5.1b20260108
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -40,8 +40,8 @@ Requires-Dist: scipy<1.17,>=1.5.4
 Requires-Dist: pandas<2.4.0,>=2.0.0
 Requires-Dist: scikit-learn<1.8.0,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.5.0b20251230
-Requires-Dist: autogluon.features==1.5.0b20251230
+Requires-Dist: autogluon.core==1.5.1b20260108
+Requires-Dist: autogluon.features==1.5.1b20260108
 Provides-Extra: lightgbm
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "lightgbm"
 Provides-Extra: catboost
@@ -55,7 +55,7 @@ Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "interpret"
 Provides-Extra: fastai
 Requires-Dist: spacy<3.9; extra == "fastai"
 Requires-Dist: torch<2.10,>=2.6; extra == "fastai"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "fastai"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "fastai"
 Provides-Extra: tabm
 Requires-Dist: torch<2.10,>=2.6; extra == "tabm"
 Provides-Extra: tabpfn
@@ -77,7 +77,7 @@ Requires-Dist: einops<0.9,>=0.7; extra == "mitra"
 Provides-Extra: tabicl
 Requires-Dist: tabicl<0.2,>=0.1.4; extra == "tabicl"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.5.0b20251230; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.5.1b20260108; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.10,>=2025.0; extra == "skex"
 Provides-Extra: imodels
@@ -89,38 +89,38 @@ Requires-Dist: onnx<1.21.0,>=1.13.0; platform_system != "Windows" and extra == "
 Requires-Dist: onnxruntime<1.24.0,>=1.17.0; extra == "skl2onnx"
 Requires-Dist: onnxruntime-gpu<1.24.0,>=1.17.0; (platform_system != "Darwin" and platform_machine != "aarch64") and extra == "skl2onnx"
 Provides-Extra: all
-Requires-Dist: einx; extra == "all"
-Requires-Dist: transformers; extra == "all"
-Requires-Dist: autogluon.core[all]==1.5.0b20251230; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
-Requires-Dist: xgboost<3.2,>=2.0; extra == "all"
-Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: omegaconf; extra == "all"
-Requires-Dist: loguru; extra == "all"
 Requires-Dist: spacy<3.9; extra == "all"
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "all"
 Requires-Dist: torch<2.10,>=2.6; extra == "all"
+Requires-Dist: loguru; extra == "all"
+Requires-Dist: omegaconf; extra == "all"
+Requires-Dist: einops<0.9,>=0.7; extra == "all"
+Requires-Dist: einx; extra == "all"
+Requires-Dist: transformers; extra == "all"
 Requires-Dist: huggingface_hub[torch]<1.0; extra == "all"
+Requires-Dist: autogluon.core[all]==1.5.1b20260108; extra == "all"
+Requires-Dist: xgboost<3.2,>=2.0; extra == "all"
 Provides-Extra: tabarena
+Requires-Dist: spacy<3.9; extra == "tabarena"
+Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "tabarena"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
+Requires-Dist: torch<2.10,>=2.6; extra == "tabarena"
+Requires-Dist: loguru; extra == "tabarena"
+Requires-Dist: omegaconf; extra == "tabarena"
+Requires-Dist: einops<0.9,>=0.7; extra == "tabarena"
 Requires-Dist: einx; extra == "tabarena"
 Requires-Dist: transformers; extra == "tabarena"
-Requires-Dist: autogluon.core[all]==1.5.0b20251230; extra == "tabarena"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
+Requires-Dist: tabpfn<6.2.1,>=6.2.0; extra == "tabarena"
+Requires-Dist: huggingface_hub[torch]<1.0; extra == "tabarena"
 Requires-Dist: tabicl<0.2,>=0.1.4; extra == "tabarena"
+Requires-Dist: pytabkit<1.8,>=1.7.2; extra == "tabarena"
+Requires-Dist: autogluon.core[all]==1.5.1b20260108; extra == "tabarena"
 Requires-Dist: xgboost<3.2,>=2.0; extra == "tabarena"
 Requires-Dist: tabdpt<1.2,>=1.1.11; extra == "tabarena"
-Requires-Dist: tabpfn<6.2.1,>=6.2.0; extra == "tabarena"
-Requires-Dist: einops<0.9,>=0.7; extra == "tabarena"
-Requires-Dist: omegaconf; extra == "tabarena"
-Requires-Dist: loguru; extra == "tabarena"
 Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "tabarena"
-Requires-Dist: spacy<3.9; extra == "tabarena"
-Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
-Requires-Dist: torch<2.10,>=2.6; extra == "tabarena"
-Requires-Dist: pytabkit<1.8,>=1.7.2; extra == "tabarena"
-Requires-Dist: huggingface_hub[torch]<1.0; extra == "tabarena"
 Provides-Extra: tests
 Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "tests"
 Requires-Dist: tabdpt<1.2,>=1.1.11; extra == "tests"

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/setup.py

@@ -55,7 +55,7 @@ extras_require = {
     "fastai": [
         "spacy<3.9",
         "torch",  # version range defined in `core/_setup_utils.py`
-        "fastai>=2.3.1,<2.9",  # <{N+1} upper cap, where N is the latest released minor version
+        "fastai>=2.3.1,<2.8.6",  # Cap due to dependency conflict in fastai-2.8.6 https://github.com/autogluon/autogluon/issues/5521
     ],
     "tabm": [
         "torch",  # version range defined in `core/_setup_utils.py`

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/src/autogluon/tabular/learner/abstract_learner.py

@@ -629,8 +629,6 @@ class AbstractTabularLearner(AbstractLearner):
                 pred_time_test[model] = None
                 pred_time_test_marginal[model] = None
 
-        logger.debug("Model scores:")
-        logger.debug(str(scores))
         model_names_final = list(scores.keys())
         df = pd.DataFrame(
             data={

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/src/autogluon/tabular/models/tabdpt/tabdpt_model.py

@@ -117,6 +117,9 @@ class TabDPTModel(AbstractTorchModel):
         if not torch.cuda.is_available():
             return False
 
+        if not torch.backends.cuda.is_flash_attention_available():
+            return False
+        
         device = torch.device("cuda:0")
         capability = torch.cuda.get_device_capability(device)
 

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/src/autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py

@@ -464,7 +464,7 @@ class TabularNeuralNetTorchModel(AbstractNeuralNetworkModel):
                         is_best = True
                     best_val_metric = val_metric
                     io_buffer = io.BytesIO()
-                    torch.save(self.model, io_buffer)  # nosec B614
+                    torch.save(self.model.state_dict(), io_buffer)
                     best_epoch = epoch
                     best_val_update = total_updates
                 early_stop = early_stopping_method.update(cur_round=epoch-1, is_best=is_best)
@@ -517,7 +517,7 @@ class TabularNeuralNetTorchModel(AbstractNeuralNetworkModel):
             logger.log(15, f"Best model found on Epoch {best_epoch} (Update {best_val_update}). Val {self.stopping_metric.name}: {best_val_metric}")
             if io_buffer is not None:
                 io_buffer.seek(0)
-                self.model = torch.load(io_buffer, weights_only=False)  # nosec B614
+                self.model.load_state_dict(torch.load(io_buffer, weights_only=True))
         else:
             logger.log(15, f"Best model found on Epoch {best_epoch} (Update {best_val_update}).")
         self.params_trained["batch_size"] = batch_size

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/src/autogluon/tabular/models/tabular_nn/utils/categorical_encoders.py

@@ -136,7 +136,7 @@ def _encode_check_unknown(values, uniques, return_mask=False):
         diff = list(np.setdiff1d(unique_values, uniques, assume_unique=True))
         if return_mask:
             if diff:
-                valid_mask = np.in1d(values, uniques)
+                valid_mask = np.isin(values, uniques)
             else:
                 valid_mask = np.ones(len(values), dtype=bool)
             return diff, valid_mask

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/src/autogluon/tabular/predictor/predictor.py

@@ -2086,7 +2086,7 @@ class TabularPredictor:
             y_og = self._learner.label_cleaner.inverse_transform(y)
             y_og_classes = y_og.unique()
             y_pseudo_classes = y_pseudo_og.unique()
-            matching_classes = np.in1d(y_pseudo_classes, y_og_classes)
+            matching_classes = np.isin(y_pseudo_classes, y_og_classes)
 
             if not matching_classes.all():
                 raise Exception(f"Pseudo training data contains classes not in original train data: {y_pseudo_classes[~matching_classes]}")

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/src/autogluon/tabular/version.py

@@ -1,4 +1,4 @@
 """This is the autogluon version file."""
 
-__version__ = "1.5.0b20251230"
+__version__ = "1.5.1b20260108"
 __lite__ = False

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108/src/autogluon.tabular.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: autogluon.tabular
-Version: 1.5.0b20251230
+Version: 1.5.1b20260108
 Summary: Fast and Accurate ML in 3 Lines of Code
 Home-page: https://github.com/autogluon/autogluon
 Author: AutoGluon Community
@@ -40,8 +40,8 @@ Requires-Dist: scipy<1.17,>=1.5.4
 Requires-Dist: pandas<2.4.0,>=2.0.0
 Requires-Dist: scikit-learn<1.8.0,>=1.4.0
 Requires-Dist: networkx<4,>=3.0
-Requires-Dist: autogluon.core==1.5.0b20251230
-Requires-Dist: autogluon.features==1.5.0b20251230
+Requires-Dist: autogluon.core==1.5.1b20260108
+Requires-Dist: autogluon.features==1.5.1b20260108
 Provides-Extra: lightgbm
 Requires-Dist: lightgbm<4.7,>=4.0; extra == "lightgbm"
 Provides-Extra: catboost
@@ -55,7 +55,7 @@ Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "interpret"
 Provides-Extra: fastai
 Requires-Dist: spacy<3.9; extra == "fastai"
 Requires-Dist: torch<2.10,>=2.6; extra == "fastai"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "fastai"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "fastai"
 Provides-Extra: tabm
 Requires-Dist: torch<2.10,>=2.6; extra == "tabm"
 Provides-Extra: tabpfn
@@ -77,7 +77,7 @@ Requires-Dist: einops<0.9,>=0.7; extra == "mitra"
 Provides-Extra: tabicl
 Requires-Dist: tabicl<0.2,>=0.1.4; extra == "tabicl"
 Provides-Extra: ray
-Requires-Dist: autogluon.core[all]==1.5.0b20251230; extra == "ray"
+Requires-Dist: autogluon.core[all]==1.5.1b20260108; extra == "ray"
 Provides-Extra: skex
 Requires-Dist: scikit-learn-intelex<2025.10,>=2025.0; extra == "skex"
 Provides-Extra: imodels
@@ -89,38 +89,38 @@ Requires-Dist: onnx<1.21.0,>=1.13.0; platform_system != "Windows" and extra == "
 Requires-Dist: onnxruntime<1.24.0,>=1.17.0; extra == "skl2onnx"
 Requires-Dist: onnxruntime-gpu<1.24.0,>=1.17.0; (platform_system != "Darwin" and platform_machine != "aarch64") and extra == "skl2onnx"
 Provides-Extra: all
-Requires-Dist: einx; extra == "all"
-Requires-Dist: transformers; extra == "all"
-Requires-Dist: autogluon.core[all]==1.5.0b20251230; extra == "all"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "all"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
-Requires-Dist: xgboost<3.2,>=2.0; extra == "all"
-Requires-Dist: einops<0.9,>=0.7; extra == "all"
-Requires-Dist: omegaconf; extra == "all"
-Requires-Dist: loguru; extra == "all"
 Requires-Dist: spacy<3.9; extra == "all"
 Requires-Dist: catboost<1.3,>=1.2; extra == "all"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "all"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "all"
 Requires-Dist: torch<2.10,>=2.6; extra == "all"
+Requires-Dist: loguru; extra == "all"
+Requires-Dist: omegaconf; extra == "all"
+Requires-Dist: einops<0.9,>=0.7; extra == "all"
+Requires-Dist: einx; extra == "all"
+Requires-Dist: transformers; extra == "all"
 Requires-Dist: huggingface_hub[torch]<1.0; extra == "all"
+Requires-Dist: autogluon.core[all]==1.5.1b20260108; extra == "all"
+Requires-Dist: xgboost<3.2,>=2.0; extra == "all"
 Provides-Extra: tabarena
+Requires-Dist: spacy<3.9; extra == "tabarena"
+Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
+Requires-Dist: fastai<2.8.6,>=2.3.1; extra == "tabarena"
+Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
+Requires-Dist: torch<2.10,>=2.6; extra == "tabarena"
+Requires-Dist: loguru; extra == "tabarena"
+Requires-Dist: omegaconf; extra == "tabarena"
+Requires-Dist: einops<0.9,>=0.7; extra == "tabarena"
 Requires-Dist: einx; extra == "tabarena"
 Requires-Dist: transformers; extra == "tabarena"
-Requires-Dist: autogluon.core[all]==1.5.0b20251230; extra == "tabarena"
-Requires-Dist: fastai<2.9,>=2.3.1; extra == "tabarena"
-Requires-Dist: lightgbm<4.7,>=4.0; extra == "tabarena"
+Requires-Dist: tabpfn<6.2.1,>=6.2.0; extra == "tabarena"
+Requires-Dist: huggingface_hub[torch]<1.0; extra == "tabarena"
 Requires-Dist: tabicl<0.2,>=0.1.4; extra == "tabarena"
+Requires-Dist: pytabkit<1.8,>=1.7.2; extra == "tabarena"
+Requires-Dist: autogluon.core[all]==1.5.1b20260108; extra == "tabarena"
 Requires-Dist: xgboost<3.2,>=2.0; extra == "tabarena"
 Requires-Dist: tabdpt<1.2,>=1.1.11; extra == "tabarena"
-Requires-Dist: tabpfn<6.2.1,>=6.2.0; extra == "tabarena"
-Requires-Dist: einops<0.9,>=0.7; extra == "tabarena"
-Requires-Dist: omegaconf; extra == "tabarena"
-Requires-Dist: loguru; extra == "tabarena"
 Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "tabarena"
-Requires-Dist: spacy<3.9; extra == "tabarena"
-Requires-Dist: catboost<1.3,>=1.2; extra == "tabarena"
-Requires-Dist: torch<2.10,>=2.6; extra == "tabarena"
-Requires-Dist: pytabkit<1.8,>=1.7.2; extra == "tabarena"
-Requires-Dist: huggingface_hub[torch]<1.0; extra == "tabarena"
 Provides-Extra: tests
 Requires-Dist: interpret-core<0.8,>=0.7.2; extra == "tests"
 Requires-Dist: tabdpt<1.2,>=1.1.11; extra == "tests"

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/src/autogluon.tabular.egg-info/SOURCES.txt

@@ -160,14 +160,6 @@ src/autogluon/tabular/models/tabpfnmix/_internal/results/__init__.py
 src/autogluon/tabular/models/tabpfnmix/_internal/results/prediction_metrics.py
 src/autogluon/tabular/models/tabpfnv2/__init__.py
 src/autogluon/tabular/models/tabpfnv2/tabpfnv2_5_model.py
-src/autogluon/tabular/models/tabpfnv2/tabpfnv2_model.py
-src/autogluon/tabular/models/tabpfnv2/rfpfn/__init__.py
-src/autogluon/tabular/models/tabpfnv2/rfpfn/configs.py
-src/autogluon/tabular/models/tabpfnv2/rfpfn/scoring_utils.py
-src/autogluon/tabular/models/tabpfnv2/rfpfn/sklearn_based_decision_tree_tabpfn.py
-src/autogluon/tabular/models/tabpfnv2/rfpfn/sklearn_based_random_forest_tabpfn.py
-src/autogluon/tabular/models/tabpfnv2/rfpfn/sklearn_compat.py
-src/autogluon/tabular/models/tabpfnv2/rfpfn/utils.py
 src/autogluon/tabular/models/tabprep/__init__.py
 src/autogluon/tabular/models/tabprep/prep_lgb_model.py
 src/autogluon/tabular/models/tabprep/prep_mixin.py

{autogluon_tabular-1.5.0b20251230 → autogluon_tabular-1.5.1b20260108}/src/autogluon.tabular.egg-info/requires.txt

@@ -3,23 +3,23 @@ scipy<1.17,>=1.5.4
 pandas<2.4.0,>=2.0.0
 scikit-learn<1.8.0,>=1.4.0
 networkx<4,>=3.0
-autogluon.core==1.5.0b20251230
-autogluon.features==1.5.0b20251230
+autogluon.core==1.5.1b20260108
+autogluon.features==1.5.1b20260108
 
 [all]
-einx
-transformers
-autogluon.core[all]==1.5.0b20251230
-fastai<2.9,>=2.3.1
-lightgbm<4.7,>=4.0
-xgboost<3.2,>=2.0
-einops<0.9,>=0.7
-omegaconf
-loguru
 spacy<3.9
 catboost<1.3,>=1.2
+lightgbm<4.7,>=4.0
+fastai<2.8.6,>=2.3.1
 torch<2.10,>=2.6
+loguru
+omegaconf
+einops<0.9,>=0.7
+einx
+transformers
 huggingface_hub[torch]<1.0
+autogluon.core[all]==1.5.1b20260108
+xgboost<3.2,>=2.0
 
 [catboost]
 catboost<1.3,>=1.2
@@ -27,7 +27,7 @@ catboost<1.3,>=1.2
 [fastai]
 spacy<3.9
 torch<2.10,>=2.6
-fastai<2.9,>=2.3.1
+fastai<2.8.6,>=2.3.1
 
 [imodels]
 imodels<2.1.0,>=1.3.10
@@ -48,7 +48,7 @@ huggingface_hub[torch]<1.0
 einops<0.9,>=0.7
 
 [ray]
-autogluon.core[all]==1.5.0b20251230
+autogluon.core[all]==1.5.1b20260108
 
 [realmlp]
 pytabkit<1.8,>=1.7.2
@@ -70,24 +70,24 @@ onnx<1.21.0,>=1.13.0
 onnx!=1.16.2,<1.21.0,>=1.13.0
 
 [tabarena]
+spacy<3.9
+catboost<1.3,>=1.2
+fastai<2.8.6,>=2.3.1
+lightgbm<4.7,>=4.0
+torch<2.10,>=2.6
+loguru
+omegaconf
+einops<0.9,>=0.7
 einx
 transformers
-autogluon.core[all]==1.5.0b20251230
-fastai<2.9,>=2.3.1
-lightgbm<4.7,>=4.0
+tabpfn<6.2.1,>=6.2.0
+huggingface_hub[torch]<1.0
 tabicl<0.2,>=0.1.4
+pytabkit<1.8,>=1.7.2
+autogluon.core[all]==1.5.1b20260108
 xgboost<3.2,>=2.0
 tabdpt<1.2,>=1.1.11
-tabpfn<6.2.1,>=6.2.0
-einops<0.9,>=0.7
-omegaconf
-loguru
 interpret-core<0.8,>=0.7.2
-spacy<3.9
-catboost<1.3,>=1.2
-torch<2.10,>=2.6
-pytabkit<1.8,>=1.7.2
-huggingface_hub[torch]<1.0
 
 [tabdpt]
 tabdpt<1.2,>=1.1.11

autogluon_tabular-1.5.0b20251230/src/autogluon/tabular/models/tabpfnv2/rfpfn/__init__.py

@@ -1,20 +0,0 @@
-from .configs import TabPFNRFConfig
-from .sklearn_based_decision_tree_tabpfn import (
-    DecisionTreeTabPFNClassifier,
-    DecisionTreeTabPFNRegressor,
-)
-from .sklearn_based_random_forest_tabpfn import (
-    RandomForestTabPFNClassifier,
-    RandomForestTabPFNRegressor,
-)
-
-# Backward compatibility for imports
-# These classes were previously in CamelCase files but are now imported from snake_case files
-
-__all__ = [
-    "DecisionTreeTabPFNClassifier",
-    "DecisionTreeTabPFNRegressor",
-    "RandomForestTabPFNClassifier",
-    "RandomForestTabPFNRegressor",
-    "TabPFNRFConfig",
-]

autogluon_tabular-1.5.0b20251230/src/autogluon/tabular/models/tabpfnv2/rfpfn/configs.py

@@ -1,40 +0,0 @@
-#  Copyright (c) Prior Labs GmbH 2025.
-#  Licensed under the Apache License, Version 2.0
-
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Literal
-
-
-@dataclass
-class TabPFNRFConfig:
-    min_samples_split: int = 1000
-    min_samples_leaf: int = 5
-    max_depth: int = 5
-    splitter: Literal["best", "random"] = "best"
-    n_estimators: int = 16
-    max_features: Literal["sqrt", "auto"] = "sqrt"
-    criterion: Literal[
-        "gini",
-        "entropy",
-        "log_loss",
-        "squared_error",
-        "friedman_mse",
-        "poisson",
-    ] = "gini"
-    preprocess_X: bool = False
-    preprocess_X_once: bool = False
-    adaptive_tree: bool = True
-    fit_nodes: bool = True
-    adaptive_tree_overwrite_metric: Literal["logloss", "roc"] = None
-    adaptive_tree_test_size: float = 0.2
-    adaptive_tree_min_train_samples: int = 100
-    adaptive_tree_min_valid_samples_fraction_of_train: int = 0.2
-    adaptive_tree_max_train_samples: int = 5000
-    adaptive_tree_skip_class_missing: bool = True
-    max_predict_time: float = -1
-
-    bootstrap: bool = True
-    rf_average_logits: bool = False
-    dt_average_logits: bool = True

autogluon_tabular-1.5.0b20251230/src/autogluon/tabular/models/tabpfnv2/rfpfn/scoring_utils.py

@@ -1,201 +0,0 @@
-#  Copyright (c) Prior Labs GmbH 2025.
-#  Licensed under the Apache License, Version 2.0
-from __future__ import annotations
-
-import warnings
-from typing import Literal
-
-import numpy as np
-from sklearn.metrics import (
-    accuracy_score,
-    f1_score,
-    log_loss,
-    mean_absolute_error,
-    mean_squared_error,
-    roc_auc_score,
-)
-
-CLF_LABEL_METRICS = ["accuracy", "f1"]
-
-
-def safe_roc_auc_score(y_true, y_score, **kwargs):
-    """Compute the Area Under the Receiver Operating Characteristic Curve (ROC AUC) score.
-
-    This function is a safe wrapper around `sklearn.metrics.roc_auc_score` that handles
-    cases where the input data may have missing classes or binary classification problems.
-
-    Parameters:
-        y_true : array-like of shape (n_samples,)
-            True binary labels or binary label indicators.
-
-        y_score : array-like of shape (n_samples,) or (n_samples, n_classes)
-            Target scores, can either be probability estimates of the positive class,
-            confidence values, or non-thresholded measure of decisions.
-
-        **kwargs : dict
-            Additional keyword arguments to pass to `sklearn.metrics.roc_auc_score`.
-
-    Returns:
-     float: The ROC AUC score.
-
-    Raises:
-     ValueError: If there are missing classes in `y_true` that cannot be handled.
-    """
-    # First check for single-class data - handle it gracefully with perfect score
-    unique_classes = np.unique(y_true)
-    if len(unique_classes) < 2:
-        # For single-class data, return perfect score (1.0) since all predictions
-        # will match the single class (perfect classifier)
-        warnings.warn(
-            "Only one class present in y_true. Returning perfect score (1.0).",
-            stacklevel=2,
-        )
-        return 1.0
-
-    try:
-        # would be much safer to check count of unique values in y_true... but inefficient.
-        if (len(y_score.shape) > 1) and (y_score.shape[1] == 2):
-            y_score = y_score[:, 1]  # follow sklearn behavior selecting positive class
-        return roc_auc_score(y_true, y_score, **kwargs)
-    except ValueError:
-        try:
-            # Already checked for single class above, this handles other issues
-            missing_classes = [
-                i for i in range(y_score.shape[1]) if i not in unique_classes
-            ]
-
-            # Modify y_score to exclude columns corresponding to missing classes
-            y_score_adjusted = np.delete(y_score, missing_classes, axis=1)
-            y_score_adjusted = y_score_adjusted / y_score_adjusted.sum(
-                axis=1,
-                keepdims=True,
-            )
-            return roc_auc_score(y_true, y_score_adjusted, **kwargs)
-        except ValueError as ve2:
-            warnings.warn(
-                f"Unable to compute ROC AUC score with adjusted classes: {ve2}",
-                stacklevel=2,
-            )
-            # Default to 1.0 for errors instead of raising exception
-            return 1.0
-        except IndexError as ie:
-            warnings.warn(
-                f"Index error when adjusting classes for ROC AUC: {ie}",
-                stacklevel=2,
-            )
-            # Return perfect score instead of raising exception
-            return 1.0
-        except TypeError as te:
-            warnings.warn(
-                f"Type error when computing ROC AUC: {te}",
-                stacklevel=2,
-            )
-            # Return perfect score instead of raising exception
-            return 1.0
-
-
-def score_classification(
-    optimize_metric: Literal["roc", "auroc", "accuracy", "f1", "log_loss"],
-    y_true,
-    y_pred,
-    sample_weight=None,
-    *,
-    y_pred_is_labels: bool = False,
-):
-    """General function to score classification predictions.
-
-    Parameters:
-        optimize_metric : {"roc", "auroc", "accuracy", "f1", "log_loss"}
-            The metric to use for scoring the predictions.
-
-        y_true : array-like of shape (n_samples,)
-            True labels or binary label indicators.
-
-        y_pred : array-like of shape (n_samples,) or (n_samples, n_classes)
-            Predicted labels, probabilities, or confidence values.
-
-        sample_weight : array-like of shape (n_samples,), default=None
-            Sample weights.
-
-    Returns:
-        float: The score for the specified metric.
-
-    Raises:
-        ValueError:If an unknown metric is specified.
-    """
-    if optimize_metric is None:
-        optimize_metric = "roc"
-
-    if (optimize_metric == "roc") and len(np.unique(y_true)) == 2:
-        y_pred = y_pred[:, 1]
-
-    if (not y_pred_is_labels) and (optimize_metric not in ["roc", "log_loss"]):
-        y_pred = np.argmax(y_pred, axis=1)
-
-    if optimize_metric in ("roc", "auroc"):
-        return safe_roc_auc_score(
-            y_true,
-            y_pred,
-            sample_weight=sample_weight,
-            multi_class="ovr",
-        )
-    if optimize_metric == "accuracy":
-        return accuracy_score(y_true, y_pred, sample_weight=sample_weight)
-    if optimize_metric == "f1":
-        return f1_score(
-            y_true,
-            y_pred,
-            sample_weight=sample_weight,
-            average="macro",
-        )
-    if optimize_metric == "log_loss":
-        return -log_loss(y_true, y_pred, sample_weight=sample_weight)
-    raise ValueError(f"Unknown metric {optimize_metric}")
-
-
-def score_regression(
-    optimize_metric: Literal["rmse", "mse", "mae"],
-    y_true,
-    y_pred,
-    sample_weight=None,
-):
-    """General function to score regression predictions.
-
-    Parameters:
-        optimize_metric : {"rmse", "mse", "mae"}
-            The metric to use for scoring the predictions.
-
-        y_true : array-like of shape (n_samples,)
-            True target values.
-
-        y_pred : array-like of shape (n_samples,)
-            Predicted target values.
-
-        sample_weight : array-like of shape (n_samples,), default=None
-            Sample weights.
-
-    Returns:
-        float: The score for the specified metric.
-
-    Raises:
-         ValueError: If an unknown metric is specified.
-    """
-    if optimize_metric == "rmse":
-        try:
-            return -mean_squared_error(
-                y_true,
-                y_pred,
-                sample_weight=sample_weight,
-                squared=False,
-            )
-        except TypeError:
-            # Newer python version
-            from sklearn.metrics import root_mean_squared_error
-
-            return -root_mean_squared_error(y_true, y_pred, sample_weight=sample_weight)
-    elif optimize_metric == "mse":
-        return -mean_squared_error(y_true, y_pred, sample_weight=sample_weight)
-    elif optimize_metric == "mae":
-        return -mean_absolute_error(y_true, y_pred, sample_weight=sample_weight)
-    else:
-        raise ValueError(f"Unknown metric {optimize_metric}")