fix space eval

This commit is contained in:
littletomatodonkey 2021-01-07 08:10:01 +00:00
parent e7bf6c86cc
commit f5008a0a97
1 changed files with 2 additions and 0 deletions

View File

@ -26,6 +26,8 @@ class RecMetric(object):
all_num = 0
norm_edit_dis = 0.0
for (pred, pred_conf), (target, _) in zip(preds, labels):
pred = pred.replace(" ", "")
target = target.replace(" ", "")
norm_edit_dis += Levenshtein.distance(pred, target) / max(
len(pred), len(target))
if pred == target: