Fix typo in loglik computation during VEstep, mentioned in #91

PLN-team · Aug 8, 2022 · e154799 · e154799
1 parent c686d7b
commit e154799
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/optim_full_cov.cpp b/src/optim_full_cov.cpp
@@ -162,7 +162,7 @@ Rcpp::List cpp_optimize_vestep_full(
     // Element-wise log-likelihood
     arma::mat Z = O + X * Theta.t() + M;
     arma::mat A = exp(Z + 0.5 * S2);
-    arma::mat loglik = sum(Y % Z - A + 0.5 * log(S2) - 0.5 * ((M * Omega) % M + S * diagmat(Omega)), 1) +
+    arma::mat loglik = sum(Y % Z - A + 0.5 * log(S2) - 0.5 * ((M * Omega) % M + S2 * diagmat(Omega)), 1) +
                        0.5 * real(log_det(Omega)) + ki(Y);
 
     return Rcpp::List::create(