src/models/__init__.py

"""Model implementations - all share a common base class."""

from .base import FraudModel
from .xgb_model import XGBFraudModel
from .lgbm_model import LGBMFraudModel
from .catboost_model import CatBoostFraudModel
from .isolation_forest_model import IsolationForestFraudModel
from .autoencoder import AutoencoderFraudModel
from .ensemble import StackingFraudEnsemble

__all__ = [
    "FraudModel",
    "XGBFraudModel",
    "LGBMFraudModel",
    "CatBoostFraudModel",
    "IsolationForestFraudModel",
    "AutoencoderFraudModel",
    "StackingFraudEnsemble",
]