another bug

chung · Jan 20, 2016 · b89b057 · b89b057
1 parent 0a6bd4f
commit b89b057
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/rmn.cpp b/src/rmn.cpp
@@ -101,8 +101,11 @@ void calcSumZ(const arma::vec& z, const arma::vec& empty, const arma::uvec& s, a
       sumZ(w, s(i)) += empty(i);
       sumZ(arma::span(w - 1 - i, w - 1), s(i)) += z(arma::span(0, i));
     }
-
-  }
+    for (; i < L; i++){
+      sumZ(w, s(i)) += empty(i);
+      sumZ(arma::span(0, w - 1), s(i)) += z(arma::span(i - w + 1, i));
+    }
+}
 
 
 void calcLnEmPN(const arma::umat& u, const arma::uvec& s, const arma::mat& mu, arma::vec& lnEmPN){