ProStock/src/training/__init__.py

"""训练模块 - ProStock 量化投资框架

提供模型训练、数据处理和预测的完整流程。
"""

# 基础抽象类
from src.training.components.base import BaseModel, BaseProcessor

# 注册中心
from src.training.registry import (
    ModelRegistry,
    ProcessorRegistry,
    register_model,
    register_processor,
)

# 数据划分器
from src.training.components.splitters import DateSplitter

# 股票池选择器配置
from src.training.components.selectors import (
    MarketCapSelectorConfig,
    StockFilterConfig,
)

# 数据处理器
from src.training.components.processors import (
    CrossSectionalStandardScaler,
    NullFiller,
    StandardScaler,
    Winsorizer,
)

# 模型
from src.training.components.models import LightGBMModel

# 数据过滤器
from src.training.components.filters import BaseFilter, STFilter

# 训练核心
from src.training.core import StockPoolManager, Trainer

# 配置
from src.training.config import TrainingConfig

__all__ = [
    # 基础抽象类
    "BaseModel",
    "BaseProcessor",
    # 注册中心
    "ModelRegistry",
    "ProcessorRegistry",
    "register_model",
    "register_processor",
    # 数据划分器
    "DateSplitter",
    # 股票池选择器配置
    "StockFilterConfig",
    "MarketCapSelectorConfig",
    # 数据处理器
    "NullFiller",
    "StandardScaler",
    "CrossSectionalStandardScaler",
    "Winsorizer",
    # 数据过滤器
    "BaseFilter",
    "STFilter",
    # 模型
    "LightGBMModel",
    # 训练核心
    "StockPoolManager",
    "Trainer",
    # 配置
    "TrainingConfig",
]