fixed deprecated function call in PCA-layers.

delve-team · Mar 27, 2023 · 79657ee · 79657ee
1 parent 04ef70f
commit 79657ee
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/delve/pca_layers.py b/delve/pca_layers.py
@@ -198,8 +198,9 @@ def _compute_autorcorrelation(self) -> torch.Tensor:
         return cov_mtx
 
     def _compute_eigenspace(self):
-        self.eigenvalues.data, self.eigenvectors.data = self._compute_autorcorrelation(
-        ).symeig(True)  #.type(self.data_dtype)
+        self.eigenvalues.data, self.eigenvectors.data = torch.linalg.eigh(
+            self._compute_autorcorrelation(),  UPLO='U'
+        )  #.type(self.data_dtype)
         self.eigenvalues.data, idx = self.eigenvalues.sort(descending=True)
         # correct numerical error, matrix must be positivly semi-definitie
         self.eigenvalues[self.eigenvalues < 0] = 0