Spaces:
Running
Running
File size: 1,056 Bytes
e17f3ba | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | """
Model Wrappers Package
======================
Provides sklearn-compatible wrappers for all benchmarking models.
Available Models:
- SAP RPT-1 (sap_rpt1_wrapper)
- TabPFN (tabpfn_wrapper)
- TabICL (tabicl_wrapper)
- AutoGluon (autogluon_wrapper)
- XGBoost (baseline_wrappers)
- CatBoost (baseline_wrappers)
- LightGBM (baseline_wrappers)
All models implement the sklearn API:
- fit(X, y)
- predict(X)
- predict_proba(X) # for classification
"""
from .base_wrapper import BaseModelWrapper
from .sap_rpt1_wrapper import SAPRPT1Wrapper
from .sap_rpt1_hf_wrapper import SAPRPT1HFWrapper
from .tabpfn_wrapper import TabPFNWrapper
from .tabicl_wrapper import TabICLWrapper
from .autogluon_wrapper import AutoGluonWrapper
from .baseline_wrappers import XGBoostWrapper, CatBoostWrapper, LightGBMWrapper
__all__ = [
'BaseModelWrapper',
'SAPRPT1Wrapper',
'SAPRPT1HFWrapper',
'TabPFNWrapper',
'TabICLWrapper',
'AutoGluonWrapper',
'XGBoostWrapper',
'CatBoostWrapper',
'LightGBMWrapper'
]
__version__ = '1.0.0'
|