ShopTRAINING/server/trainers/__init__.py

"""
药店销售预测系统 - 模型训练模块
"""

from .mlstm_trainer import train_product_model_with_mlstm
from .kan_trainer import train_product_model_with_kan
from .tcn_trainer import train_product_model_with_tcn
from .transformer_trainer import train_product_model_with_transformer
from .xgboost_trainer import train_product_model_with_xgboost

# 默认训练函数
from .mlstm_trainer import train_product_model_with_mlstm as train_product_model

__all__ = [
    'train_product_model',
    'train_product_model_with_mlstm',
    'train_product_model_with_kan',
    'train_product_model_with_tcn',
    'train_product_model_with_transformer',
    'train_product_model_with_xgboost'
]