From 771fc7579c323c870d5bb1ce285e66807f8e7a97 Mon Sep 17 00:00:00 2001 From: Mikhail Erofeev Date: Mon, 12 Sep 2016 23:26:26 +0300 Subject: [PATCH] Fix bug in Hidden Markov Model final probabilities computation. --- seqlearn/hmm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/seqlearn/hmm.py b/seqlearn/hmm.py index 59df882..b7dfacb 100644 --- a/seqlearn/hmm.py +++ b/seqlearn/hmm.py @@ -64,7 +64,7 @@ def fit(self, X, y, lengths): init_prob = np.log(Y[start].sum(axis=0) + alpha) init_prob -= logsumexp(init_prob) - final_prob = np.log(Y[start].sum(axis=0) + alpha) + final_prob = np.log(Y[end - 1].sum(axis=0) + alpha) final_prob -= logsumexp(final_prob) feature_prob = np.log(safe_sparse_dot(Y.T, X) + alpha)