Spaces:
Sleeping
Sleeping
fix: Evaluator gt
Browse filesFormer-commit-id: 50ea9dd03bdf8019e2303e9f82076ae9c78d1769
- evaluation/evaluation.py +3 -3
evaluation/evaluation.py
CHANGED
@@ -20,14 +20,14 @@ class Evaluator:
|
|
20 |
for (pred_s, gt_s) in aligned_srt:
|
21 |
print("pred_s.source_text: ", pred_s.source_text)
|
22 |
print("pred_s.translation: ", pred_s.translation)
|
23 |
-
print("gt_s.source_text: ", gt_s.
|
24 |
|
25 |
-
scores_dict = scorer.get_scores(pred_s.source_text, pred_s.translation, gt_s.
|
26 |
print("scores_dict: ", scores_dict)
|
27 |
|
28 |
scores_dict['Source'] = pred_s.source_text
|
29 |
scores_dict['Prediction'] = pred_s.translation
|
30 |
-
scores_dict['Ground Truth'] = gt_s.
|
31 |
result_data.append(scores_dict)
|
32 |
|
33 |
eval_df = pd.DataFrame(result_data)
|
|
|
20 |
for (pred_s, gt_s) in aligned_srt:
|
21 |
print("pred_s.source_text: ", pred_s.source_text)
|
22 |
print("pred_s.translation: ", pred_s.translation)
|
23 |
+
print("gt_s.source_text: ", gt_s.source_text)
|
24 |
|
25 |
+
scores_dict = scorer.get_scores(pred_s.source_text, pred_s.translation, gt_s.source_text)
|
26 |
print("scores_dict: ", scores_dict)
|
27 |
|
28 |
scores_dict['Source'] = pred_s.source_text
|
29 |
scores_dict['Prediction'] = pred_s.translation
|
30 |
+
scores_dict['Ground Truth'] = gt_s.source_text
|
31 |
result_data.append(scores_dict)
|
32 |
|
33 |
eval_df = pd.DataFrame(result_data)
|