models/__init__.py

"""One sub-package per classifier, plus the registry that collects them."""

from models import logistic_regression, naive_bayes, random_forest, svm

__all__ = [
    "build_models_dict",
    "logistic_regression",
    "naive_bayes",
    "random_forest",
    "svm",
]


def build_models_dict():
    from models.registry import build_models_dict as _build_models_dict

    return _build_models_dict()