From 274fc5b45bcf641ec8d30c2bc362bcab88aca5da Mon Sep 17 00:00:00 2001 From: wangkaifu Date: Fri, 6 May 2022 16:13:32 +0800 Subject: [PATCH] fix bug: NPTag postprocess --- paddlenlp/taskflow/knowledge_mining.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/paddlenlp/taskflow/knowledge_mining.py b/paddlenlp/taskflow/knowledge_mining.py index 5295785ac013..05f6b0507f7a 100644 --- a/paddlenlp/taskflow/knowledge_mining.py +++ b/paddlenlp/taskflow/knowledge_mining.py @@ -760,11 +760,11 @@ def _postprocess(self, inputs): if cls_label_can in self._name_dict: result['label'] = cls_label_can break - else: - labels_can = self._tree.search_similar_word(cls_label) - if len(labels_can) != 0: - result['label'] = labels_can[0][0] - break + else: + labels_can = self._tree.search_similar_word(cls_label) + if len(labels_can) != 0: + result['label'] = labels_can[0][0] + break if self._linking: if result['label'] in self._name_dict: result['category'] = self._name_dict[result['label']]