sum->mean

ppegolo · Sep 18, 2024 · b2fe641 · b2fe641
1 parent 2de0b26
commit b2fe641
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sportran/md/maxlike.py b/sportran/md/maxlike.py
@@ -351,7 +351,9 @@ def log_likelihood_wishart(
         # log_pdf = (
         #     0.5*(coeff_detV*np.log(detV + eps) + coeff_detX*np.log(detX + eps) - trinvV_X) - logG
         # )
-        tot = -np.sum(log_pdf)
+
+        # return the negative LL because of `minimize`
+        tot = -np.mean(log_pdf)
         return tot
 
     def log_likelihood_complex_wishart(