Merge pull request #112 from go-inoue/fix_bert_scoring

Fixed missing diac error.
CAMeL-Lab · Mar 6, 2023 · 194be5e · 194be5e
2 parents 794d18e + 7b5b900
commit 194be5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/camel_tools/disambig/bert/unfactored.py b/camel_tools/disambig/bert/unfactored.py
@@ -470,7 +470,7 @@ def _scored_analyses(self, word_dd, prediction):
             # return the predictions from BERT
             return [ScoredAnalysis(0,                      # score
                                    bert_analysis,          # analysis
-                                   bert_analysis['diac'],  # diac
+                                   word_dd,                # diac
                                    -99,                    # pos_lex_logprob
                                    -99,                    # lex_logprob
                                    )]