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

fixed other CNN references

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
NN/models/cnn_model_pytorch.py
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()
}

11
NN/models/enhanced_cnn_with_orderbook.py
View File

@ -501,7 +501,16 @@ class EnhancedCNNWithOrderBook(nn.Module):
# Get probabilities
q_values = outputs['q_values']
probs = F.softmax(q_values, dim=1)
confidence = outputs['confidence'].item()
# Handle confidence shape properly to avoid scalar conversion errors
confidence_tensor = outputs['confidence']
if isinstance(confidence_tensor, torch.Tensor):
if confidence_tensor.numel() == 1:
confidence = confidence_tensor.item()
else:
confidence = confidence_tensor.flatten()[0].item()
else:
confidence = float(confidence_tensor)
# Action selection with confidence thresholding
if confidence >= self.confidence_threshold: