refactor: 存储层迁移DuckDB + 模块重构
- 存储层重构: HDF5 → DuckDB(UPSERT模式、线程安全存储) - Sync类迁移: DataSync从sync.py迁移到api_daily.py(职责分离) - 模型模块重构: src/models → src/pipeline(更清晰的命名) - 新增因子模块: factors/momentum (MA、收益率排名)、factors/financial - 新增API接口: api_namechange、api_bak_basic - 新增训练入口: training模块(main.py、pipeline配置) - 工具函数统一: get_today_date等移至utils.py - 文档更新: AGENTS.md添加架构变更历史
This commit is contained in:
27
src/training/main.py
Normal file
27
src/training/main.py
Normal file
@@ -0,0 +1,27 @@
|
||||
"""训练流程入口脚本
|
||||
|
||||
运行方式:
|
||||
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)
|
||||
Reference in New Issue
Block a user