#!/bin/bash
# A股超短期选股系统 - 自动训练脚本
# 用法: bash scripts/auto_train.sh
# 建议cron: 0 16 * * 5 cd /root/nginx/gupiao && bash scripts/auto_train.sh >> logs/train.log 2>&1

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
LOG_DIR="$PROJECT_DIR/logs"
mkdir -p "$LOG_DIR"

echo "========================================"
echo "  自动训练开始: $(date '+%Y-%m-%d %H:%M:%S')"
echo "========================================"

cd "$PROJECT_DIR"

python3 train.py --stocks 300 --days 250

echo ""
echo "训练完成: $(date '+%Y-%m-%d %H:%M:%S')"
echo "========================================"