Bugfix

ClaudioZeni · May 13, 2021 · 1913ba0 · 1913ba0
1 parent bd195e9
commit 1913ba0
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/raffy/linear_potential.py b/raffy/linear_potential.py
@@ -136,8 +136,6 @@ def fit(self, X, Y=None, Y_en=None, noise=1e-8,
         gtg = np.einsum('na, nb -> ab', g_tot, g_tot)
         # Add regularization
         noise = self.noise*np.ones(len(gtg))
-        if Y_en is not None:
-            noise[-len(X):] = noise[-len(X):]/10
         gtg[np.diag_indices_from(gtg)] += noise
         # Cholesky Decomposition to find alpha
         L_ = cholesky(gtg, lower=True)