From ffbf5a9f59e1702f24dd2bc5853373da6af4f0ce Mon Sep 17 00:00:00 2001 From: Yener Date: Wed, 4 Dec 2019 09:41:47 +0800 Subject: [PATCH] fix S bug --- jiagu/analyze.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/jiagu/analyze.py b/jiagu/analyze.py index 58508de..29ab5bd 100644 --- a/jiagu/analyze.py +++ b/jiagu/analyze.py @@ -80,7 +80,9 @@ class Analyze(object): words.append(tmp_word) tmp_word = "" else: - tmp_word = "" + if tmp_word != '': + words.append(tmp_word) + tmp_word = "" words.append(w) if tmp_word: words.append(tmp_word) @@ -89,7 +91,9 @@ class Analyze(object): def cws_text(self, sentence): if sentence == '': return [''] - labels = self.seg_model.predict(list(sentence)) + + sentence = list(sentence) + labels = self.seg_model.predict(sentence) return self.__lab2word(sentence, labels) def seg(self, sentence):