50 lines
1.1 KiB
Batchfile
50 lines
1.1 KiB
Batchfile
@echo off
|
|
echo ============================================================
|
|
echo Neural Network Trading System - PyTorch Implementation
|
|
echo ============================================================
|
|
|
|
call conda activate gpt-gpu
|
|
|
|
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 ============================================================ |