This commit is contained in:
ddrwode
2026-02-20 21:37:52 +08:00
parent 45a56e0aef
commit 29c1b3ffd1
5 changed files with 24 additions and 1 deletions

21
run_scheme_b_train.py Normal file
View File

@@ -0,0 +1,21 @@
#!/usr/bin/env python
"""
仅运行方案BLightGBM训练并保存模型供实盘脚本 get_live_signal 使用。
需先保证 models/database.db 中有 15m/5m/1h K 线(例如运行 抓取多周期K线.py
"""
import argparse
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parent))
from strategy.ai_strategy import run_ai_strategy
if __name__ == '__main__':
p = argparse.ArgumentParser(description='方案B 训练并保存模型(实盘用)')
p.add_argument('--period', type=int, default=15, help='主周期分钟,默认 15')
p.add_argument('--start', type=str, default=None, help='回测开始日期,如 2024-01-01')
p.add_argument('--end', type=str, default=None, help='回测结束日期,如 2024-12-31')
args = p.parse_args()
run_ai_strategy(model_type='lightgbm', period=args.period,
start_date=args.start, end_date=args.end)