Correct use of index_max() result

eddelbuettel · Nov 22, 2024 · a09ae52 · a09ae52
1 parent a8181b0
commit a09ae52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mlpack/methods/hmm/hmm_impl.hpp b/src/mlpack/methods/hmm/hmm_impl.hpp
@@ -529,7 +529,7 @@ double HMM<Distribution>::Predict(const arma::mat& dataSeq,
     {
       arma::vec prob = logStateProb.col(t - 1) + logTransition.row(j).t();
       index = prob.index_max();
-      logStateProb(j, t) = index + logProbs(t, j);
+      logStateProb(j, t) = prob[index] + logProbs(t, j);
       stateSeqBack(j, t) = index;
     }
   }