models/__init__.py

# __init__.py
__all__ = ["cross_entropy_model", "classifier", "triplet_loss_model"]

from .cross_entropy_model import *
from .classifier import *
from .triplet_loss_model import *