This commit is contained in:
Dobromir Popov 2025-03-29 03:57:04 +02:00
parent ebbc0ed2d7
commit 43803caaf1
2 changed files with 30 additions and 1 deletions

25
.gitignore vendored
View File

@ -17,4 +17,27 @@ models/trading_agent_final.pt.backup
logs/
trade_logs/
*.csv
cache/
cache/
realtime_chart.log
training_results.png
training_stats.csv
__pycache__/realtime.cpython-312.pyc
cache/BTC_USDT_1d_candles.csv
cache/BTC_USDT_1h_candles.csv
cache/BTC_USDT_1m_candles.csv
cache/ETH_USDT_1d_candles.csv
cache/ETH_USDT_1h_candles.csv
cache/ETH_USDT_1m_candles.csv
models/trading_agent_best_pnl.pt
models/trading_agent_best_reward.pt
models/trading_agent_final.pt
NN/__pycache__/realtime_main.cpython-312.pyc
NN/__pycache__/realtime-main.cpython-312.pyc
NN/models/__pycache__/__init__.cpython-312.pyc
NN/models/__pycache__/cnn_model_pytorch.cpython-312.pyc
NN/models/__pycache__/cnn_model.cpython-312.pyc
NN/models/__pycache__/transformer_model_pytorch.cpython-312.pyc
NN/utils/__pycache__/data_interface.cpython-312.pyc
NN/utils/__pycache__/multi_data_interface.cpython-312.pyc
NN/utils/__pycache__/realtime_analyzer.cpython-312.pyc
models/trading_agent_best_pnl.pt

View File

@ -134,6 +134,12 @@ class CNNModelPyTorch:
output_size (int): Size of the output (1 for regression, 3 for classification)
timeframes (list): List of timeframes used (for logging)
"""
# Action tracking
self.action_counts = {
'BUY': 0,
'SELL': 0,
'HOLD': 0
}
self.window_size = window_size
self.num_features = num_features
self.output_size = output_size