new nn wip

This commit is contained in:
Dobromir Popov
2025-03-25 13:38:25 +02:00
parent 50eb50696b
commit 0042581275
18 changed files with 3358 additions and 294 deletions

50
run_pytorch_nn.bat Normal file
View 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 ============================================================