new nn wip
This commit is contained in:
50
run_pytorch_nn.bat
Normal file
50
run_pytorch_nn.bat
Normal file
@@ -0,0 +1,50 @@
|
||||
@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 ============================================================
|
||||
Reference in New Issue
Block a user