fix divide zero error

2026-09-24 23:33:08 +08:00 · 2021-02-02 15:45:57 +08:00
parent 23df5bcf12
commit dc0a21ffc7
1 changed files with 1 additions and 1 deletions
@@ -29,7 +29,7 @@ class RecMetric(object):
            pred = pred.replace(" ", "")
            target = target.replace(" ", "")
            norm_edit_dis += Levenshtein.distance(pred, target) / max(
-                len(pred), len(target))
+                len(pred), len(target), 1)
            if pred == target:
                correct_num += 1
            all_num += 1