From cd94e692417c927b9fc1bc90c9df1d7db38b6009 Mon Sep 17 00:00:00 2001 From: fxsjy Date: Mon, 8 Oct 2012 21:27:01 +0800 Subject: [PATCH] fix a bug --- jieba/finalseg/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/jieba/finalseg/__init__.py b/jieba/finalseg/__init__.py index 7680e95..f4ce1c4 100644 --- a/jieba/finalseg/__init__.py +++ b/jieba/finalseg/__init__.py @@ -27,7 +27,7 @@ def viterbi(obs, states, start_p, trans_p, emit_p): newpath[y] = path[state] + [y] path = newpath - (prob, state) = max([(V[len(obs) - 1][y], y) for y in states]) + (prob, state) = max([(V[len(obs) - 1][y], y) for y in ('E','S')]) return (prob, path[state]) @@ -35,6 +35,7 @@ def viterbi(obs, states, start_p, trans_p, emit_p): def __cut(sentence): prob, pos_list = viterbi(sentence,('B','M','E','S'), prob_start, prob_trans, prob_emit) begin, next = 0,0 + #print pos_list, sentence for i,char in enumerate(sentence): pos = pos_list[i] if pos=='B':