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