@echo off
echo ============================================================
echo Neural Network Trading System - PyTorch Implementation
echo ============================================================

call conda activate gpt-gpu

REM Install missing dependencies if needed
echo Checking for required packages...
python -c "import sklearn" 2>NUL
if %ERRORLEVEL% NEQ 0 (
    echo Installing scikit-learn...
    call conda install -y scikit-learn
)

REM Parse command-line arguments
set MODE=train
set MODEL_TYPE=cnn
set SYMBOL=BTC/USDT
set EPOCHS=100

:parse
if "%~1"=="" goto endparse
if /i "%~1"=="--mode" (
    set MODE=%~2
    shift
    shift
    goto parse
)
if /i "%~1"=="--model" (
    set MODEL_TYPE=%~2
    shift
    shift
    goto parse
)
if /i "%~1"=="--symbol" (
    set SYMBOL=%~2
    shift
    shift
    goto parse
)
if /i "%~1"=="--epochs" (
    set EPOCHS=%~2
    shift
    shift
    goto parse
)
shift
goto parse
:endparse

echo Running Neural Network in %MODE% mode with %MODEL_TYPE% model for %SYMBOL% for %EPOCHS% epochs

python -m NN.main --mode %MODE% --symbol %SYMBOL% --timeframes 1h 4h --window-size 20 --output-size 3 --batch-size 32 --epochs %EPOCHS% --model-type %MODEL_TYPE% --framework pytorch

echo ============================================================
echo Run completed.
echo ============================================================