increase prediction horizon
This commit is contained in:
@@ -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
|
||||
|
Reference in New Issue
Block a user