increase prediction horizon

This commit is contained in:
Dobromir Popov
2025-09-09 09:50:14 +03:00
parent 34780d62c7
commit 2e1b3be2cd
5 changed files with 670 additions and 6 deletions

View File

@@ -140,14 +140,15 @@ class EnhancedCNNModel(nn.Module):
- Large capacity for complex pattern learning
"""
def __init__(self,
def __init__(self,
input_size: int = 60,
feature_dim: int = 50,
output_size: int = 2, # BUY/SELL for 2-action system
base_channels: int = 256, # Increased from 128 to 256
num_blocks: int = 12, # Increased from 6 to 12
num_attention_heads: int = 16, # Increased from 8 to 16
dropout_rate: float = 0.2):
dropout_rate: float = 0.2,
prediction_horizon: int = 1): # New: Prediction horizon in minutes
super().__init__()
self.input_size = input_size