fix rouge score

wanli2014 · Jul 6, 2023 · 559fe75 · 559fe75
1 parent 687b7ef
commit 559fe75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils/seq2seq.py b/src/utils/seq2seq.py
@@ -48,7 +48,7 @@ def __call__(self, eval_preds: Sequence[Union[np.ndarray, Tuple[np.ndarray]]]) -
             hypothesis = list(jieba.cut(pred))
             reference = list(jieba.cut(label))
 
-            if len(" ".join(hypothesis).split()) == 0:
+            if len(" ".join(hypothesis).split()) == 0 or len(" ".join(reference).split()) == 0:
                 result = {"rouge-1": {"f": 0.0}, "rouge-2": {"f": 0.0}, "rouge-l": {"f": 0.0}}
             else:
                 rouge = Rouge()