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,
)

__all__ = [
    # 基础抽象类
    "BaseModel",
    "BaseProcessor",
    # 注册中心
    "ModelRegistry",
    "ProcessorRegistry",
    "register_model",
    "register_processor",
]