stability

This commit is contained in:
Dobromir Popov
2025-07-22 21:18:31 +03:00
parent 6c91bf0b93
commit 153ebe6ec2
3 changed files with 37 additions and 8 deletions

View File

@ -230,7 +230,8 @@ class TradingOrchestrator:
self.model_states['dqn']['checkpoint_loaded'] = True
self.model_states['dqn']['checkpoint_filename'] = metadata.checkpoint_id
checkpoint_loaded = True
logger.info(f"DQN checkpoint loaded: {metadata.checkpoint_id} (loss={metadata.loss:.4f})")
loss_str = f"{metadata.loss:.4f}" if metadata.loss is not None else "N/A"
logger.info(f"DQN checkpoint loaded: {metadata.checkpoint_id} (loss={loss_str})")
except Exception as e:
logger.warning(f"Error loading DQN checkpoint: {e}")
@ -269,7 +270,8 @@ class TradingOrchestrator:
self.model_states['cnn']['checkpoint_loaded'] = True
self.model_states['cnn']['checkpoint_filename'] = metadata.checkpoint_id
checkpoint_loaded = True
logger.info(f"CNN checkpoint loaded: {metadata.checkpoint_id} (loss={metadata.loss:.4f})")
loss_str = f"{metadata.loss:.4f}" if metadata.loss is not None else "N/A"
logger.info(f"CNN checkpoint loaded: {metadata.checkpoint_id} (loss={loss_str})")
except Exception as e:
logger.warning(f"Error loading CNN checkpoint: {e}")
@ -356,7 +358,8 @@ class TradingOrchestrator:
self.model_states['cob_rl']['checkpoint_loaded'] = True
self.model_states['cob_rl']['checkpoint_filename'] = metadata.checkpoint_id
checkpoint_loaded = True
logger.info(f"COB RL checkpoint loaded: {metadata.checkpoint_id} (loss={metadata.loss:.4f})")
loss_str = f"{metadata.loss:.4f}" if metadata.loss is not None else "N/A"
logger.info(f"COB RL checkpoint loaded: {metadata.checkpoint_id} (loss={loss_str})")
except Exception as e:
logger.warning(f"Error loading COB RL checkpoint: {e}")
@ -547,7 +550,7 @@ class TradingOrchestrator:
if self.cob_integration:
try:
logger.info("Attempting to start COB integration...")
await self.cob_integration.start_streaming()
await self.cob_integration.start()
logger.info("COB Integration streaming started successfully.")
except Exception as e:
logger.error(f"Failed to start COB integration streaming: {e}")

View File

@ -114,6 +114,34 @@ class EnhancedRealtimeTrainingSystem:
logger.info("Enhanced Real-time Training System initialized")
def _get_dqn_state_features(self, symbol: str) -> Optional[np.ndarray]:
"""Get DQN state features from orchestrator"""
try:
if not self.orchestrator:
return None
# Get DQN state from orchestrator if available
if hasattr(self.orchestrator, 'build_comprehensive_rl_state'):
rl_state = self.orchestrator.build_comprehensive_rl_state(symbol)
if rl_state and 'state_vector' in rl_state:
return np.array(rl_state['state_vector'], dtype=np.float32)
# Fallback: create basic state from available data
if len(self.real_time_data['ohlcv_1m']) > 0:
latest_bar = self.real_time_data['ohlcv_1m'][-1]
basic_state = [
latest_bar.get('close', 0) / 10000.0, # Normalized price
latest_bar.get('volume', 0) / 1000000.0, # Normalized volume
0.0, 0.0, 0.0 # Placeholder features
]
return np.array(basic_state, dtype=np.float32)
return None
except Exception as e:
logger.debug(f"Error getting DQN state features for {symbol}: {e}")
return None
def start_training(self):
"""Start the enhanced real-time training system"""
if self.is_training:
@ -1885,7 +1913,7 @@ class EnhancedRealtimeTrainingSystem:
and self.orchestrator.rl_agent):
# Use RL agent to make prediction
current_state = self._get_dqn_state(symbol)
current_state = self._get_dqn_state_features(symbol)
if current_state is None:
return
action = self.orchestrator.rl_agent.act(current_state, explore=False)

View File

@ -5205,9 +5205,7 @@ class CleanTradingDashboard:
if self.orchestrator and hasattr(self.orchestrator, 'add_decision_callback'):
def connect_worker():
try:
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
loop.run_until_complete(self.orchestrator.add_decision_callback(self._on_trading_decision))
self.orchestrator.add_decision_callback(self._on_trading_decision)
logger.info("Successfully connected to orchestrator for trading signals.")
except Exception as e:
logger.error(f"Orchestrator connection worker failed: {e}")