fixed other CNN references

This commit is contained in:
Dobromir Popov
2025-06-24 21:13:06 +03:00
parent 8685319989
commit e7ea17b626
3 changed files with 49 additions and 17 deletions

View File

@ -431,11 +431,11 @@ class EnhancedCNNModel(nn.Module):
return {
'action': action,
'action_name': 'BUY' if action == 0 else 'SELL',
'confidence': float(confidence),
'confidence': confidence, # Already converted to float above
'action_confidence': action_confidence,
'probabilities': probs.tolist(),
'regime_probabilities': regime.tolist(),
'volatility_prediction': float(volatility),
'volatility_prediction': volatility, # Already converted to float above
'raw_logits': outputs['logits'].cpu().numpy()[0].tolist()
}