models/__init__.py

from .lstm import LSTM
from .transformer import Transformer
from .xgboost import Xgboost
from .cnn import CNN