popov/gogo2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix cob data errors in T model

Browse Source
This commit is contained in:
Dobromir Popov
2025-11-11 13:51:02 +02:00
parent a6bd5f64ff
commit 4c04503f3e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
NN/models/advanced_transformer_trading.py
View File

@@ -1182,9 +1182,9 @@ class TradingTransformerTrainer:
price_data_1h=batch.get('price_data_1h'), price_data_1h=batch.get('price_data_1h'),
price_data_1d=batch.get('price_data_1d'), price_data_1d=batch.get('price_data_1d'),
btc_data_1m=batch.get('btc_data_1m'), btc_data_1m=batch.get('btc_data_1m'),
cob_data=batch['cob_data'], cob_data=batch.get('cob_data'), # Use .get() to handle missing key
tech_data=batch['tech_data'], tech_data=batch.get('tech_data'),
market_data=batch['market_data'], market_data=batch.get('market_data'),
position_state=batch.get('position_state'), position_state=batch.get('position_state'),
price_data=batch.get('price_data') # Legacy fallback price_data=batch.get('price_data') # Legacy fallback
) )