pinnnn commited on
Commit
dca0a7c
1 Parent(s): 1b862fc

fix: Evaluator gt

Browse files

Former-commit-id: 50ea9dd03bdf8019e2303e9f82076ae9c78d1769

Files changed (1) hide show
  1. 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.translation)
24
 
25
- scores_dict = scorer.get_scores(pred_s.source_text, pred_s.translation, gt_s.translation)
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.translation
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)