training fixes

This commit is contained in:
Dobromir Popov
2025-06-26 14:18:04 +03:00
parent b7ccd0f97b
commit 1f47576723
5 changed files with 291 additions and 36 deletions

View File

@ -168,7 +168,10 @@ class TradingOrchestrator:
# Initialize CNN Model
try:
from NN.models.enhanced_cnn import EnhancedCNN
self.cnn_model = EnhancedCNN()
# CNN model expects input_shape and n_actions parameters
cnn_input_shape = self.config.cnn.get('input_shape', 100)
cnn_n_actions = self.config.cnn.get('n_actions', 3)
self.cnn_model = EnhancedCNN(input_shape=cnn_input_shape, n_actions=cnn_n_actions)
# Load best checkpoint and capture initial state
if hasattr(self.cnn_model, 'load_best_checkpoint'):
@ -275,6 +278,7 @@ class TradingOrchestrator:
except Exception as e:
logger.error(f"Error initializing COB integration: {e}")
logger.info("COB integration will be disabled - dashboard will run with basic price data")
self.cob_integration = None
async def _start_cob_integration(self):