fix bigru crf offset index error (#3418)

PaddlePaddle · Oct 9, 2022 · 2dfdfa0 · 2dfdfa0
1 parent c297ab0
commit 2dfdfa0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/test_tipc/bigru_crf/data.py b/tests/test_tipc/bigru_crf/data.py
@@ -141,7 +141,7 @@ def parse_result(words, preds, lengths, word_vocab, label_vocab):
             for index in words[sent_index][:lengths[sent_index]]
         ]
         tags = [
-            id2label_dict[index]
+            id2label_dict.get(index, 'O')
             for index in preds[sent_index][:lengths[sent_index]]
         ]
 

diff --git a/tests/test_tipc/bigru_crf/deploy/predict.py b/tests/test_tipc/bigru_crf/deploy/predict.py
@@ -108,7 +108,7 @@ def parse_result(words, preds, lengths, word_vocab, label_vocab):
             for index in words[sent_index][:lengths[sent_index]]
         ]
         tags = [
-            id2label_dict[index]
+            id2label_dict.get(index, 'O')
             for index in preds[sent_index][:lengths[sent_index]]
         ]