Merge pull request #112 from go-inoue/fix_bert_scoring

owo · web-flow · commit 194be5e8a440 · 2023-03-06T15:50:10.000+04:00
Fixed missing diac error.
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
                                    )]
