20 lines
589 B
Python
20 lines
589 B
Python
"""
|
|
药店销售预测系统 - 模型训练模块
|
|
"""
|
|
|
|
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 .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'
|
|
]
|