autogluon.tabular 1.3.2b20250610__py3-none-any.whl → 1.4.1b20251214__py3-none-any.whl

{autogluon.tabular-1.3.2b20250610.dist-info → autogluon_tabular-1.4.1b20251214.dist-info}/RECORD

@@ -1,13 +1,15 @@
-autogluon.tabular-1.3.2b20250610-py3.9-nspkg.pth,sha256=cQGwpuGPqg1GXscIwt-7PmME1OnSpD-7ixkikJ31WAY,554
+autogluon.tabular-1.4.1b20251214-py3.11-nspkg.pth,sha256=kAlKxjI5mE3Pwwqphu2maN5OBQk8W8ew70e_qbI1c6A,482
 autogluon/tabular/__init__.py,sha256=2OXpJCvENRHubBTYNIPpHX93WWuFZzsJBtTZbNVHVas,400
-autogluon/tabular/version.py,sha256=-UKiYXc8KZBEbFRj_FU-zL8tLS7VqGTS4xEMGv_SM7M,91
+autogluon/tabular/version.py,sha256=CYNrmn5KfprHt9fJxfHv56jQZ5Q-BPbHMgM1kWYZDD8,91
 autogluon/tabular/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/configs/config_helper.py,sha256=JsdVGmpcYL88GPKBznPtqJ1sGaByOSvLn7KWU-HyVoQ,21085
+autogluon/tabular/configs/config_helper.py,sha256=Rby5gRhuY5IlZWdKbtsmzbSt948B97qxwQ2f1MbH_38,21070
 autogluon/tabular/configs/feature_generator_presets.py,sha256=EV5Ym8VW15q92MwOUpTi7wZFS2QooM51fLg3RdUsn-M,1223
-autogluon/tabular/configs/hyperparameter_configs.py,sha256=hp8J7g5GY3Couz929f1ItawobCw-isLTZJBcLoJY348,18035
-autogluon/tabular/configs/presets_configs.py,sha256=2Jlq1X9sVmVlyUxWsZpDV7ma2TncH5Y2HXDML7x2gYc,6810
+autogluon/tabular/configs/hyperparameter_configs.py,sha256=aQ1rrF8P0MX4Ic5M33O96JtKV-K7YpDrgJmWhYmEyug,6848
+autogluon/tabular/configs/pipeline_presets.py,sha256=ccrT3C56pYHW8x8VB_Q9zAu_eCxlgNQpt7TXpVUzDfE,4761
+autogluon/tabular/configs/presets_configs.py,sha256=_C9wTfKVRyoomtYa04RqNyw1CEOYc_5Q3QKejqDp754,7674
 autogluon/tabular/configs/zeroshot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/configs/zeroshot/zeroshot_portfolio_2023.py,sha256=oKO_2nEpI_EiLaUGmNN-3kPBIp5ATndbCOaVZ1m0048,29911
+autogluon/tabular/configs/zeroshot/zeroshot_portfolio_2023.py,sha256=6yd84vPqOk-6sLCoM_e_PlphrR2NZUjliS7L1SMKMug,29777
+autogluon/tabular/configs/zeroshot/zeroshot_portfolio_2025.py,sha256=NXwfqZLQLx4kdvRqF6deFDdhZZKxbfgpUurdB0kqOh8,11996
 autogluon/tabular/experimental/__init__.py,sha256=PpkdMSv_pPZted1XRIuzcFWKjM-66VMUukTnCcoiW0s,100
 autogluon/tabular/experimental/_scikit_mixin.py,sha256=cKeCmtURAXZnhQGrkCBw5rmACCQF7biAWTT3qX8bM2Q,2281
 autogluon/tabular/experimental/_tabular_classifier.py,sha256=7lGoFdvkHiZS3VpcXo97q4ENV9qyIVDExlWkm0wzL3s,2527
@@ -16,27 +18,32 @@ autogluon/tabular/experimental/plot_leaderboard.py,sha256=BN_kB-zmOZNUYWyI7z9pF6
 autogluon/tabular/learner/__init__.py,sha256=Hhmk5WpKQHohVmI-veOaKMelKJpIdzeXrmw_DPn3DTU,63
 autogluon/tabular/learner/abstract_learner.py,sha256=0kf0huvg0nphe-lrdKtNTzdIFr14jzJPsfZDRBkKo3g,55253
 autogluon/tabular/learner/default_learner.py,sha256=hjdKbcFtIQxQ3-k1LiGOo-w5sLxIIQAyFLs3-R35aw0,24781
-autogluon/tabular/models/__init__.py,sha256=fZDKUKiD9hDzEyFXXbt7_b4yADK9peREdP8QoukWukQ,1036
+autogluon/tabular/models/__init__.py,sha256=grZ23UfuNZ_LxoNdl-yjIUmq71TeovT5CJPhbatiqvg,1252
 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
-autogluon/tabular/models/automm/ft_transformer.py,sha256=yZ9-TTA4GbtutHhz0Djkrl-rIFNxc7A2LBOFOXYOxVY,3886
+autogluon/tabular/models/automm/automm_model.py,sha256=MoydDuPEd5atbUPlVDzWLTKLB7EchcPdSVVncxA9jEM,11355
+autogluon/tabular/models/automm/ft_transformer.py,sha256=X-IEi5uKme7SoRcHnPjGTByzrjCB85I7RpB0hS36TLQ,3897
 autogluon/tabular/models/catboost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/catboost/callbacks.py,sha256=l8x17n_w7oEFs-iDECSdBKZ89yW5g1z-zvj4XLgQPkw,7098
-autogluon/tabular/models/catboost/catboost_model.py,sha256=Dv62XDuJ_sFvU95xeY1gHuV_F2RoYDGaFa6-4XNwnnU,17849
+autogluon/tabular/models/catboost/callbacks.py,sha256=QvyiynQoxjvfYaYwGNSF5N3gc_wqI9mi1nQiawL0EJ4,7194
+autogluon/tabular/models/catboost/catboost_model.py,sha256=tAT_eklRJDARJsbS72-Nn8PxLmKgIvffzjjrTI1XMXM,18041
 autogluon/tabular/models/catboost/catboost_softclass_utils.py,sha256=UiW0SUb3hFueW5qYtQn6Sbk7Wg7BWN4jqKWeFtbMvgU,3919
-autogluon/tabular/models/catboost/catboost_utils.py,sha256=YSc94V4DjrwbmkeUM8306zV7z21oq-K-qGCOj0UE_wg,3167
+autogluon/tabular/models/catboost/catboost_utils.py,sha256=zJMIsbgyW_JH0eULhUeu_TWR0Qfmf34CnED7c7NvXBw,3899
 autogluon/tabular/models/catboost/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/catboost/hyperparameters/parameters.py,sha256=Hxi4mPTc2ML9GdpW0TalkDgtsYJLwpEcd-LiyLOsmlA,956
 autogluon/tabular/models/catboost/hyperparameters/searchspaces.py,sha256=Oe86ixuvd1xJCdSHs2Oh5Ifx0501YJBsdyL2l9Z4nxM,1458
+autogluon/tabular/models/ebm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/tabular/models/ebm/ebm_model.py,sha256=PyocCEPxByB-E5gRCZitI5gsP6DVYlxmRx8bbZ31guA,8524
+autogluon/tabular/models/ebm/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/tabular/models/ebm/hyperparameters/parameters.py,sha256=IbDv3Ufx8CGHvejqSbAggZKlMq5X9k0Ggclm_DCoiII,1080
+autogluon/tabular/models/ebm/hyperparameters/searchspaces.py,sha256=G6zgHERKt_KJlVfZ06tFKw2aOUuM7DdDyCm0s5RBXoc,2191
 autogluon/tabular/models/fastainn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/fastainn/callbacks.py,sha256=3WvOEwqd1YAVInooKsFOTzAkCLeIXjEelsglYwfofq0,4788
 autogluon/tabular/models/fastainn/fastai_helpers.py,sha256=gGYzyrAFl8hi8GnsemZNLGZn5xr7cyJXdFl08PIlza4,1393
 autogluon/tabular/models/fastainn/imports_helper.py,sha256=ICxA8ty47-oZu0Q9AjKCQe8uVi340Iu0NFruxvJPrbA,330
 autogluon/tabular/models/fastainn/quantile_helpers.py,sha256=d89GKvSRBgOy9EqcDI83MK5sqPRxP6JJ3BmPLmKnB0o,1808
-autogluon/tabular/models/fastainn/tabular_nn_fastai.py,sha256=wtvs2VclaEvt-DDtTAxteVpDTXTAYPBp72jGxlGQwgE,29522
+autogluon/tabular/models/fastainn/tabular_nn_fastai.py,sha256=FqT6xqhU2XoTWJ0yY_ZmT3JI6ranl63vpdPkn6JFbos,29666
 autogluon/tabular/models/fastainn/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/fastainn/hyperparameters/parameters.py,sha256=DkQwAZZ7CuODKoljr-yrkx-uFxBSPRxkKuvPdwO-UhQ,2069
 autogluon/tabular/models/fastainn/hyperparameters/searchspaces.py,sha256=5qdknZDrHtdPdrhSqjamYQrCxvupXvlN3bVGEPgs48E,1660
@@ -50,34 +57,67 @@ autogluon/tabular/models/imodels/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 autogluon/tabular/models/imodels/imodels_models.py,sha256=89uQwbRAtqcUvPwYsKnER8SUMIbwkGZUd9spoG_mP10,4878
 autogluon/tabular/models/knn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/knn/_knn_loo_variants.py,sha256=-n2znYS7OBA0bZvtei6JZiEMRWp4GX-Qp64uheaHyhQ,4562
-autogluon/tabular/models/knn/knn_model.py,sha256=qZwBnDVPT2Fd5aDPZqcwugszeUYREm5nZ_-bHVVE3_s,13977
+autogluon/tabular/models/knn/knn_model.py,sha256=I7wPRy38oD03f_3KN7Q_CyoJJucDPrPQyJqjgovmx8Q,14061
 autogluon/tabular/models/knn/knn_rapids_model.py,sha256=0FFApNZFH8nyrDqlBSUV7jO-2fLe0-h_UHp1GsyQJ8E,1550
 autogluon/tabular/models/knn/knn_utils.py,sha256=XU1cxVXp1BAoQnja2_KmSIn9_q9gZkjAya7-9b0uStk,7455
 autogluon/tabular/models/lgb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/lgb/callbacks.py,sha256=KJB1KmebA88qHT206KSfvm5NamGuv5lRzy7O9dOwW-M,12243
-autogluon/tabular/models/lgb/lgb_model.py,sha256=yCCro6D--vyQSFQBow9jnsUhbAp5iRkSV0TcLrPgqQM,25756
+autogluon/tabular/models/lgb/lgb_model.py,sha256=kRIcBBIDMJ2inaZeJXO5uhAG0qUigwYseJoFQ7jzqQE,27415
 autogluon/tabular/models/lgb/lgb_utils.py,sha256=jzTDTzP-z7gcBGZyy1_0YkyTOLbU5DLeRqtil4FCZPI,7382
 autogluon/tabular/models/lgb/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/lgb/hyperparameters/parameters.py,sha256=LLEQ-Ns3HElWBsFJx3ogRV7L6qw_nXlcl7EyO0C0fVQ,1336
 autogluon/tabular/models/lgb/hyperparameters/searchspaces.py,sha256=tvNNR7niWz_B-PndYQXb6vVNABxSfBYRHj6ZVQJ1x2E,1930
 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_model.py,sha256=2A6e8Itw-PgjOLjVXeo8bJwFQuVSGYwJNVxhHxFQXlw,15732
+autogluon/tabular/models/lr/lr_preprocessing_utils.py,sha256=tgb75V6zHfMJh8m9GDs5404ItdfwNakqykTk0qjBtFE,1045
+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
+autogluon/tabular/models/mitra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/tabular/models/mitra/mitra_model.py,sha256=TzjozU19zQLU09S2tM8Sfe7TiTBSDDjld-tVt5L1JGQ,13954
+autogluon/tabular/models/mitra/sklearn_interface.py,sha256=vyg8kkmYKzEJRWiehEqEsgZeOCV20tnZAZaaaJkwDuA,17739
+autogluon/tabular/models/mitra/_internal/__init__.py,sha256=dN2dz1pGMgQTFiSf9oYbyq23iJUxV8QNlOX3qw3KUO4,35
+autogluon/tabular/models/mitra/_internal/config/__init__.py,sha256=Exu_Sx6-K-D5peDQ_TibsjZpqAALs2-9IXfq8hu1mwU,40
+autogluon/tabular/models/mitra/_internal/config/config_pretrain.py,sha256=CeaD96EcDX69LdcLTYGlFmYLdBNINEJXRMWmJ6LbhTg,6038
+autogluon/tabular/models/mitra/_internal/config/config_run.py,sha256=CVna6KOwmF-rIxcyH3mHm63jvM1C6RdFbRLgUGEXDn0,677
+autogluon/tabular/models/mitra/_internal/config/enums.py,sha256=hlyhgXHvHZKgYK1z3DHSHxEsuCHOE7Y2AdokjOG8SWs,3930
+autogluon/tabular/models/mitra/_internal/core/__init__.py,sha256=hgy4uzJfTQFt9hVlbSrOZU9LSUbLM-uZUnG04f1CUcs,31
+autogluon/tabular/models/mitra/_internal/core/callbacks.py,sha256=xYkJUXiGzLvpWcj6a_wRJUK7f_zgjd1BLA8nH6Hc884,2605
+autogluon/tabular/models/mitra/_internal/core/get_loss.py,sha256=hv0t7zvyZ-DgA5PbKpbX_ayq8tEvuW_nJhbudMDqkDk,2243
+autogluon/tabular/models/mitra/_internal/core/get_optimizer.py,sha256=UgGO6lduVZTKZmYAmE207o2Dqs4e3_hyzaoSOQ0iK6A,3412
+autogluon/tabular/models/mitra/_internal/core/get_scheduler.py,sha256=2lzdAxDOYZNq76pmK-FjCOX5MX6cqUSMjqVu8BX9jfY,2238
+autogluon/tabular/models/mitra/_internal/core/prediction_metrics.py,sha256=fai0VnDm0mNjJzx8e1JXdB77PKQsmfbtn8zybD9_qD0,4394
+autogluon/tabular/models/mitra/_internal/core/trainer_finetune.py,sha256=tI8sN9mv3PtEBdmDxcBgzderZ7YQdtn6MxtOWAc8or8,17908
+autogluon/tabular/models/mitra/_internal/data/__init__.py,sha256=u4ZTvTQNIHqqxilkVqTmYShI2jFMCOyMdv1GRExvtj0,42
+autogluon/tabular/models/mitra/_internal/data/collator.py,sha256=o2F7ODs_eUnV947lCQTx9RugrANidCdiwnZWtdVNJnE,2300
+autogluon/tabular/models/mitra/_internal/data/dataset_finetune.py,sha256=AYxyQ1NJZ3pAp6ny-Y_hqw_4VtyW5X1AABchf7pVsSM,4340
+autogluon/tabular/models/mitra/_internal/data/dataset_split.py,sha256=0uvfyiKrzipde4ZcCDwTE1E3zHelE8xbuNvCeL38J5c,2033
+autogluon/tabular/models/mitra/_internal/data/preprocessor.py,sha256=zx2pWrpDaGSSawPaj7ieRjFOtct_Fyh8LYjo_YtlNG0,13821
+autogluon/tabular/models/mitra/_internal/models/__init__.py,sha256=K0vh5pyrntXp-o7gWNgQ0ZvDbxgeQuRgb6u8ecdjFhA,45
+autogluon/tabular/models/mitra/_internal/models/base.py,sha256=PKpMPT5OT9JFnmYPnhzFUeZPwdNM1e-k97_gW8GZq0Y,468
+autogluon/tabular/models/mitra/_internal/models/embedding.py,sha256=74O6cGWhUyHxg4-wiQwy4sPeDYQze2ekI9H5mLUtSLg,6223
+autogluon/tabular/models/mitra/_internal/models/tab2d.py,sha256=o_S572-nKrhwxmEFaDSTvTLE7KztOvQmARRrc7CIeCY,25783
+autogluon/tabular/models/mitra/_internal/utils/__init__.py,sha256=0mhykAqjMmcEc8Y2od_DMPMk8f66LZHWM7qFdUrPddU,34
+autogluon/tabular/models/mitra/_internal/utils/set_seed.py,sha256=UnXzYfhmfT_tNAofKtLkKpwB9b6HVf9cpI4mKvoBuNM,340
+autogluon/tabular/models/realmlp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/tabular/models/realmlp/realmlp_model.py,sha256=3pe_yhOGW8cbX3KgNs25s3FP0P3FzVSAS-hd4jMFjDg,14573
 autogluon/tabular/models/rf/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/rf/rf_model.py,sha256=2iG45F0k17No7ycrfVMWBfU_WqJwC4MTehaFGsguLzE,21598
+autogluon/tabular/models/rf/rf_model.py,sha256=auvNHx0qD9Pz8rS6yNIuG9cHzFNquv8fOVS7FWZNIAw,21721
 autogluon/tabular/models/rf/rf_quantile.py,sha256=2S8FE8po9lMnZaeKuVkzOUFOcdil46ZbFqm49OuvNZY,36460
 autogluon/tabular/models/rf/rf_rapids_model.py,sha256=3s-8M11dzCl_2Lu5iB3H8YjHLgyP_SElrm_4w_HfmqY,2028
 autogluon/tabular/models/rf/compilers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/rf/compilers/native.py,sha256=HhaqQRkVuf9UEEJPsHcdYCmuWBMYtyqRwwB_N2qxG2M,1313
 autogluon/tabular/models/rf/compilers/onnx.py,sha256=pvaZWdl2JJaE2pFU0mFugzhnybePqe0x1-5oLOvogA0,4318
-autogluon/tabular/models/tabpfn/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon/tabular/models/tabpfn/tabpfn_model.py,sha256=PEYMuIh5TFLIDy3hcjfz1DcvDu77rbwRq0pKWyuUR04,6787
+autogluon/tabular/models/tabicl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/tabular/models/tabicl/tabicl_model.py,sha256=_Eq3g9babdC17kyvAA0rIqtZEtiRGwM2XngkbWevXpU,6283
+autogluon/tabular/models/tabm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/tabular/models/tabm/_tabm_internal.py,sha256=fRQ-s5PN94kWqf3LRDen7su_fd-d332YKxdms30FoZM,21066
+autogluon/tabular/models/tabm/rtdl_num_embeddings.py,sha256=XssNMaUM0E0G8Grzl_VkVsLt2FcMf3I4cplfvQdVum0,30156
+autogluon/tabular/models/tabm/tabm_model.py,sha256=_SGc7R87ug9m8KGd_BgC9maJ7sjOAlYB9vtg1omwOto,13640
+autogluon/tabular/models/tabm/tabm_reference.py,sha256=byyP6lcJjA4THbP1VDTgJkj62zyz2S3mEvxWB-kFROw,21944
 autogluon/tabular/models/tabpfnmix/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabpfnmix/tabpfnmix_model.py,sha256=Bo-JMnNgI0fYXXTHy3zLAE1OHZv9ikgH4bFBaVSa79g,16174
+autogluon/tabular/models/tabpfnmix/tabpfnmix_model.py,sha256=NAuV3rJia-UNnFwiFU5tkz6vzZ2lokQ_12vUJ3E6wAA,16498
 autogluon/tabular/models/tabpfnmix/_internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabpfnmix/_internal/tabpfnmix_classifier.py,sha256=_WIO_YQBUCfprKYLHxUNEICPb5XWZw4zbw00DuiTk_s,3426
 autogluon/tabular/models/tabpfnmix/_internal/tabpfnmix_regressor.py,sha256=J6JvrK6L6y3s-Ah6sHQdjSK0mwAMP-Wy3RRBwzB0AoA,3196
@@ -102,15 +142,24 @@ autogluon/tabular/models/tabpfnmix/_internal/models/foundation/embedding.py,sha2
 autogluon/tabular/models/tabpfnmix/_internal/models/foundation/foundation_transformer.py,sha256=bhNpGIA5BKqIVX-kDW4bZLgsOB_A8iNsnpgoyyBLR98,5383
 autogluon/tabular/models/tabpfnmix/_internal/results/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabpfnmix/_internal/results/prediction_metrics.py,sha256=1tRPHyViSSLJ7BkQJi6wai-PwXJ56od86Dy1WWKWZq4,1743
+autogluon/tabular/models/tabpfnv2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+autogluon/tabular/models/tabpfnv2/tabpfnv2_model.py,sha256=nXZcq4SMV54dciOKFM57Suc9eVyXQXy-2iN6moRt2b8,14801
+autogluon/tabular/models/tabpfnv2/rfpfn/__init__.py,sha256=yE5XAhGxKEFV0JcelZ_JTQZIWGlVEVUQ9a-lxcH_Esc,585
+autogluon/tabular/models/tabpfnv2/rfpfn/configs.py,sha256=lzBY9kKOeBZACVrtRDPHF4ATs9g1rxyNnIs2CMjE20c,1175
+autogluon/tabular/models/tabpfnv2/rfpfn/scoring_utils.py,sha256=uvHsfvnnMdg4tP3_7zAilktkw7nr65LaqfVKXabXAow,6785
+autogluon/tabular/models/tabpfnv2/rfpfn/sklearn_based_decision_tree_tabpfn.py,sha256=-KQNm_HYWem6HWUsdbnIX4lKe-eW0PQAXZUny2kqego,55582
+autogluon/tabular/models/tabpfnv2/rfpfn/sklearn_based_random_forest_tabpfn.py,sha256=FRJSelTtDaKnpsKKHphjy2rJrFX302miSdHZ0YqHxCQ,28045
+autogluon/tabular/models/tabpfnv2/rfpfn/sklearn_compat.py,sha256=jv2ZHsGwcO4Inhxtol_tig3NoXZQR649dhmW_Kv69QY,29607
+autogluon/tabular/models/tabpfnv2/rfpfn/utils.py,sha256=vjMQsNaZZcW1BBf0hduSCtrNCtSd467xfkhsbHspUog,3489
 autogluon/tabular/models/tabular_nn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabular_nn/compilers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/tabular_nn/compilers/native.py,sha256=W8d8cqBj7U-KVhfGK3hdtGj8JJm3lXr_SecU0615Gbs,1330
 autogluon/tabular/models/tabular_nn/compilers/onnx.py,sha256=3mj9_5p6YMOuKbYk7FBQ2Ijhm1kGzfqq6cyyKLUKLOo,14804
 autogluon/tabular/models/tabular_nn/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabular_nn/hyperparameters/parameters.py,sha256=Z3t_U1f7jfolPey6lzqgJyoFbVgoncFNSvCKXSuLxeU,6465
+autogluon/tabular/models/tabular_nn/hyperparameters/parameters.py,sha256=kGvfuDZa9wDCCTEeytVLKhOAeR0pCcoVNJcWjketmBI,6375
 autogluon/tabular/models/tabular_nn/hyperparameters/searchspaces.py,sha256=pT9cJ3MaWPnaQwAf47Yz6f0-L9qDBknahERbggAp52U,2810
 autogluon/tabular/models/tabular_nn/torch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py,sha256=enCnQYN9yNkcTEoPirQZ-NkGbLSZdguiilGn5ASDyEE,42997
+autogluon/tabular/models/tabular_nn/torch/tabular_nn_torch.py,sha256=TGVMv_ClKh0iYVVCqgd19DE-1fXk_VODpsXIMvzI3Sw,42978
 autogluon/tabular/models/tabular_nn/torch/tabular_torch_dataset.py,sha256=RdnQGZSrvY1iuJB4JTANniH3Dorw-DP0Em_JK3_h7RM,13497
 autogluon/tabular/models/tabular_nn/torch/torch_network_modules.py,sha256=Qc3PwXTD8A7PgXi6EGuaBCrN3jsFAXDLCW7i6tE5wYI,11338
 autogluon/tabular/models/tabular_nn/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -121,7 +170,7 @@ autogluon/tabular/models/text_prediction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
 autogluon/tabular/models/text_prediction/text_prediction_v1_model.py,sha256=PBN7F98qgEAO6U76rV_hxZfAmKr_XpVKjElOdBvfX8c,1090
 autogluon/tabular/models/xgboost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/xgboost/callbacks.py,sha256=PuRQUg3AEjgvFa-dpstRFoEVM9jHDe5W4XYSdDPRqoE,7009
-autogluon/tabular/models/xgboost/xgboost_model.py,sha256=_HFwfEbAg0CllEUEk9HVLsXXVBcuC_qsGd7z7arJYPs,15220
+autogluon/tabular/models/xgboost/xgboost_model.py,sha256=tKVLvBnuTbDaFwBRVDZ5ADo4PjBF2FDR93Ib86WYTMM,15630
 autogluon/tabular/models/xgboost/xgboost_utils.py,sha256=FVqZ8h4JAe_pifSvNx83cLZHwsuzTXylrrcan07AoNo,5757
 autogluon/tabular/models/xgboost/hyperparameters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autogluon/tabular/models/xgboost/hyperparameters/parameters.py,sha256=ay6bVVpiPzftbtz6TTS76w7j4vjDjzHFpuf2Bjf6Zu4,1673
@@ -130,27 +179,27 @@ autogluon/tabular/models/xt/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 autogluon/tabular/models/xt/xt_model.py,sha256=qOHJ5h1lHI7uYJfbl0BWm-29R3MNp2WeZB9ptcq5Xis,1003
 autogluon/tabular/predictor/__init__.py,sha256=zCMgjxQlWpDWnr1l1xjBCiK3rWC3N3RoD8UXBnazT74,107
 autogluon/tabular/predictor/interpretable_predictor.py,sha256=5UeKgnMFsfY65tiO3kxfHBPr03lyswLrgdtjPhI0Y7Q,6934
-autogluon/tabular/predictor/predictor.py,sha256=j-ymKlk3O5wBHF60mh_flx2k2A1KvFAkANy32XzeYME,356173
+autogluon/tabular/predictor/predictor.py,sha256=fjw7CQALXZ7AR18ryLm4xWwDzRBeUnrmNubPS8U_pmQ,361223
 autogluon/tabular/registry/__init__.py,sha256=vZpzX4Xve7bfA9crt5LxjgQv9PPfxbi1E1U6Im0Y_xU,93
-autogluon/tabular/registry/_ag_model_registry.py,sha256=m39NrbsmKjzxAVLrzQBS2dlDJk_t0jwXRb-gM5ROyr8,1414
+autogluon/tabular/registry/_ag_model_registry.py,sha256=2Zx5qxXvOdXIbL1FKslNh2M_JM2YG_7GvsCMFF11wDY,1578
 autogluon/tabular/registry/_model_registry.py,sha256=Rl8Q7BLzaif4hxNxJF20xGE02vrWwh2ZuUaTmA-UJnE,6824
 autogluon/tabular/testing/__init__.py,sha256=XrEGLmMdmRT6QHNR13M9wna57LO4O3Q4tt27Ca8omAc,79
-autogluon/tabular/testing/fit_helper.py,sha256=gVHTdAsp_lSZ_qbwjXM7aA5fI32zHj3_zXwEXC9C_ds,19586
-autogluon/tabular/testing/generate_datasets.py,sha256=UXPNfviUNZqGcx4mTYIloJxRED6DRMxAgHqbOvjEUCs,3603
+autogluon/tabular/testing/fit_helper.py,sha256=pj3P0ENMDhr04laxsLL0_IDX-8msMFo9Wn5XSLFCaqI,21092
+autogluon/tabular/testing/generate_datasets.py,sha256=nvcAmI-tOh5fwx_ZTx2aRa1n7CsXb96wbR-xqNy1C5w,3884
 autogluon/tabular/testing/model_fit_helper.py,sha256=ZjWpw2nyeFnsrccmkfQtx3qbA8HJx282XX2rwdS-LIs,3808
 autogluon/tabular/trainer/__init__.py,sha256=PW_PGL-tWoQzx3ES2S53bQEZOtsRWTYiM9QdOqsk0dI,38
-autogluon/tabular/trainer/abstract_trainer.py,sha256=F9bT78KHyQi96FbPs6ufaswVQJBzxoK3sTXyRXXpK_g,232416
+autogluon/tabular/trainer/abstract_trainer.py,sha256=9FiBqOV2h8era6KfydFSqhTlh7RnHkvlvzqsZuij7nE,232527
 autogluon/tabular/trainer/auto_trainer.py,sha256=ZQgQKFT1iHzzun5o5ojdq5pSQmr9ctTkNhe2r9OPOr0,8731
 autogluon/tabular/trainer/model_presets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autogluon/tabular/trainer/model_presets/presets.py,sha256=3gM_QFpG_BaVFIf8T0nCd-GFMQHhRJU3WrD563AcouY,16120
+autogluon/tabular/trainer/model_presets/presets.py,sha256=hoWADaOG576Q_XLV1nY_ju1OWi7EJwHay4jjljqt_E0,16546
 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.2b20250610.dist-info/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-autogluon.tabular-1.3.2b20250610.dist-info/METADATA,sha256=ztVtInq2mUB5_aWfwJLFX9OVGnkCZOhZ57nda5_EvhE,14069
-autogluon.tabular-1.3.2b20250610.dist-info/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
-autogluon.tabular-1.3.2b20250610.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-autogluon.tabular-1.3.2b20250610.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250610.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
-autogluon.tabular-1.3.2b20250610.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-autogluon.tabular-1.3.2b20250610.dist-info/RECORD,,
+autogluon_tabular-1.4.1b20251214.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+autogluon_tabular-1.4.1b20251214.dist-info/licenses/NOTICE,sha256=7nPQuj8Kp-uXsU0S5so3-2dNU5EctS5hDXvvzzehd7E,114
+autogluon_tabular-1.4.1b20251214.dist-info/METADATA,sha256=XbmjT9lmPhMkbhK6fgfIBrv6zMe1EMZ3wvoTx_Waons,17015
+autogluon_tabular-1.4.1b20251214.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+autogluon_tabular-1.4.1b20251214.dist-info/namespace_packages.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon_tabular-1.4.1b20251214.dist-info/top_level.txt,sha256=giERA4R78OkJf2ijn5slgjURlhRPzfLr7waIcGkzYAo,10
+autogluon_tabular-1.4.1b20251214.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+autogluon_tabular-1.4.1b20251214.dist-info/RECORD,,

{autogluon.tabular-1.3.2b20250610.dist-info → autogluon_tabular-1.4.1b20251214.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.45.1)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

autogluon/tabular/models/tabpfn/__init__.py

@@ -1 +0,0 @@
-

autogluon/tabular/models/tabpfn/tabpfn_model.py

@@ -1,153 +0,0 @@
-from __future__ import annotations
-
-import numpy as np
-import pandas as pd
-
-from autogluon.core.constants import BINARY, MULTICLASS
-from autogluon.core.models import AbstractModel
-from autogluon.core.utils import generate_train_test_split
-from autogluon.features.generators import LabelEncoderFeatureGenerator
-
-
-class TabPFNModel(AbstractModel):
-    """
-    AutoGluon model wrapper to the TabPFN model: https://github.com/automl/TabPFN
-
-    Paper: "TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Second"
-    Authors: Noah Hollmann, Samuel Müller, Katharina Eggensperger, and Frank Hutter
-
-    TabPFN is a viable model option when inference speed is not a concern,
-    and the number of rows of training data is less than 10,000.
-
-    Additionally, TabPFN is only available for classification tasks with up to 10 classes and 100 features.
-
-    To use this model, `tabpfn` must be installed.
-    To install TabPFN, you can run `pip install autogluon.tabular[tabpfn]` or `pip install tabpfn`.
-    """
-    ag_key = "TABPFN"
-    ag_name = "TabPFN"
-    ag_priority = 110
-
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-        self._feature_generator = None
-
-    def _fit(self, X: pd.DataFrame, y: pd.Series, **kwargs):
-        from tabpfn import TabPFNClassifier
-
-        ag_params = self._get_ag_params()
-        sample_rows = ag_params.get("sample_rows")
-        max_features = ag_params.get("max_features")
-        max_classes = ag_params.get("max_classes")
-        if max_classes is not None and self.num_classes > max_classes:
-            # TODO: Move to earlier stage when problem_type is checked
-            raise AssertionError(f"Max allowed classes for the model is {max_classes}, " f"but found {self.num_classes} classes.")
-
-        # TODO: Make sample_rows generic
-        if sample_rows is not None and len(X) > sample_rows:
-            X, y = self._subsample_train(X=X, y=y, num_rows=sample_rows)
-        X = self.preprocess(X)
-        num_features = X.shape[1]
-        # TODO: Make max_features generic
-        if max_features is not None and num_features > max_features:
-            raise AssertionError(f"Max allowed features for the model is {max_features}, " f"but found {num_features} features.")
-        hyp = self._get_model_params()
-        N_ensemble_configurations = hyp.get("N_ensemble_configurations")
-        self.model = TabPFNClassifier(device="cpu", N_ensemble_configurations=N_ensemble_configurations).fit(  # TODO: Add GPU option
-            X, y, overwrite_warning=True
-        )
-
-    # TODO: Make this generic by creating a generic `preprocess_train` and putting this logic prior to `_preprocess`.
-    def _subsample_train(self, X: pd.DataFrame, y: pd.Series, num_rows: int, random_state=0) -> (pd.DataFrame, pd.Series):
-        num_rows_to_drop = len(X) - num_rows
-        X, _, y, _ = generate_train_test_split(
-            X=X,
-            y=y,
-            problem_type=self.problem_type,
-            test_size=num_rows_to_drop,
-            random_state=random_state,
-            min_cls_count_train=1,
-        )
-        return X, y
-
-    def _preprocess(self, X: pd.DataFrame, **kwargs) -> np.ndarray:
-        """
-        Converts categorical to label encoded integers
-        Keeps missing values, as TabPFN automatically handles missing values internally.
-        """
-        X = super()._preprocess(X, **kwargs)
-        if self._feature_generator is None:
-            self._feature_generator = LabelEncoderFeatureGenerator(verbosity=0)
-            self._feature_generator.fit(X=X)
-        if self._feature_generator.features_in:
-            X = X.copy()
-            X[self._feature_generator.features_in] = self._feature_generator.transform(X=X)
-        X = X.to_numpy(dtype=np.float32)
-        return X
-
-    def _set_default_params(self):
-        """
-        By default, we only use 1 ensemble configurations to speed up inference times.
-        Increase the value to improve model quality while linearly increasing inference time.
-
-        Model quality improvement diminishes significantly beyond `N_ensemble_configurations=8`.
-        """
-        default_params = {
-            "N_ensemble_configurations": 1,
-        }
-        for param, val in default_params.items():
-            self._set_default_param_value(param, val)
-
-    @classmethod
-    def supported_problem_types(cls) -> list[str] | None:
-        return ["binary", "multiclass"]
-
-    def _get_default_auxiliary_params(self) -> dict:
-        """
-        TabPFN was originally learned on synthetic datasets with 1024 rows, and struggles to
-        leverage additional rows effectively beyond a certain point.
-
-        In the TabPFN paper, performance appeared to stagnate around 4000 rows of training data (Figure 10).
-        Thus, we set `sample_rows=4096`, to only use that many rows of training data, even if more is available.
-
-        TODO: TabPFN scales poorly on large datasets, so we set `max_rows=20000`.
-         Not implemented yet, first move this logic to the trainer level to avoid `refit_full` edge-case crashes.
-        TabPFN only works on datasets with at most 100 features, so we set `max_features=100`.
-        TabPFN only works on datasets with at most 10 classes, so we set `max_classes=10`.
-        """
-        default_auxiliary_params = super()._get_default_auxiliary_params()
-        default_auxiliary_params.update(
-            {
-                "sample_rows": 4096,
-                # 'max_rows': 20000,
-                "max_features": 100,
-                "max_classes": 10,
-            }
-        )
-        return default_auxiliary_params
-
-    # FIXME: Enabling parallel bagging TabPFN creates a lot of warnings / potential failures from Ray
-    # TODO: Consider not setting `max_sets=1`, and only setting it in the preset hyperparameter definition.
-    @classmethod
-    def _get_default_ag_args_ensemble(cls, **kwargs) -> dict:
-        """
-        Set max_sets to 1 when bagging, otherwise inference time could become extremely slow.
-        Set fold_fitting_strategy to sequential_local, as parallel folding causing many warnings / potential errors from Ray.
-        """
-        default_ag_args_ensemble = super()._get_default_ag_args_ensemble(**kwargs)
-        extra_ag_args_ensemble = {
-            "max_sets": 1,
-            "fold_fitting_strategy": "sequential_local",
-        }
-        default_ag_args_ensemble.update(extra_ag_args_ensemble)
-        return default_ag_args_ensemble
-
-    def _ag_params(self) -> set:
-        return {"sample_rows", "max_features", "max_classes"}
-
-    def _more_tags(self) -> dict:
-        """
-        Because TabPFN doesn't use validation data for early stopping, it supports refit_full natively.
-        """
-        tags = {"can_refit_full": True}
-        return tags

autogluon.tabular-1.3.2b20250610-py3.9-nspkg.pth

@@ -1 +0,0 @@
-import sys, types, os;has_mfs = sys.version_info > (3, 5);p = os.path.join(sys._getframe(1).f_locals['sitedir'], *('autogluon',));importlib = has_mfs and __import__('importlib.util');has_mfs and __import__('importlib.machinery');m = has_mfs and sys.modules.setdefault('autogluon', importlib.util.module_from_spec(importlib.machinery.PathFinder.find_spec('autogluon', [os.path.dirname(p)])));m = m or sys.modules.setdefault('autogluon', types.ModuleType('autogluon'));mp = (m or []) and m.__dict__.setdefault('__path__',[]);(p not in mp) and mp.append(p)