bug fix in gauss-newton hessian

AaltoML · Mar 6, 2023 · 0907b6f · 0907b6f
1 parent 61cb0eb
commit 0907b6f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -161,3 +161,5 @@ bayesnewton.egg-info/*
 .idea/
 
 src/data/audio_training_textures_Heavy_rain_on_hard_surface.wav
+
+build/
diff --git a/bayesnewton/likelihoods.py b/bayesnewton/likelihoods.py
@@ -171,7 +171,7 @@ def gauss_newton(self, y, f):
         log_target = -0.5 * V.T @ V + self.log_normaliser(f)
         jacobian = -J.T @ V + ZJ
         # hessian_approx = -J.T @ J + self.log_normaliser_hessian(f)
-        hessian_approx = -J.T @ J - np.diag((ZJ**2).flattent())
+        hessian_approx = -J.T @ J - np.diag((ZJ**2).flatten())
         # second_order_term = -H.T * V
         return log_target, jacobian, hessian_approx  # , second_order_term