From 207ab70d28deffdcd34fb4c5e4caae41d0ab45e5 Mon Sep 17 00:00:00 2001
From: dsman1823 <37937854+dsman1823@users.noreply.github.com>
Date: Wed, 10 Apr 2024 23:07:54 +0200
Subject: [PATCH] Update nn.py

---
 deepdowmine/nn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deepdowmine/nn.py b/deepdowmine/nn.py
index 9e00a97..3accca7 100644
--- a/deepdowmine/nn.py
+++ b/deepdowmine/nn.py
@@ -121,7 +121,7 @@ def forward(self, x):
         ) 
         x_cov = x_cov.reshape(n_samples, 20, 5)
         
-        covmat = self.covariance_layer(x_cov, shrinkage_strategy="diagonal")
+        covmat = self.covariance_layer(x_cov)
         
         exp_rets = torch.tanh(
             self.linear(x_exp_rets)