"""训练流程入口脚本

运行方式：
    uv run python -m src.training.main

或：
    uv run python src/training/main.py
"""

from src.training.pipeline import run_training


if __name__ == "__main__":
    # 运行完整训练流程
    # 训练集：20180101 - 20230101
    # 测试集：20230101 - 20240101
    result = run_training(
        train_start="20190101",
        train_end="20250101",
        test_start="20250101",
        test_end="20260101",
        top_n=5,
        output_path="output/top_stocks.tsv",
    )

    print("\n[Result] Top stocks selection:")
    print(result)