Fix: trading_intelligence/prediction_model.py - all 174 tests passing

trading_intelligence/prediction_model.py

@@ -373,9 +373,9 @@ class MultiTaskLoss(nn.Module):
         self.alpha_risk = alpha_risk
         
         # Learned task uncertainty weights (Kendall et al.)
-        self.log_sigma_direction = nn.Parameter(torch.zeros(1))
-        self.log_sigma_return = nn.Parameter(torch.zeros(1))
-        self.log_sigma_risk = nn.Parameter(torch.zeros(1))
+        self.log_sigma_direction = nn.Parameter(torch.tensor(0.0))
+        self.log_sigma_return = nn.Parameter(torch.tensor(0.0))
+        self.log_sigma_risk = nn.Parameter(torch.tensor(0.0))
     
     def forward(self, predictions: Dict[str, torch.Tensor], 
                 targets: Dict[str, torch.Tensor]) -> Dict[str, torch.Tensor]: