lhzstar commited on
Commit
b190683
1 Parent(s): 7fba753

new commits

Browse files
Files changed (2) hide show
  1. .gitignore +1 -1
  2. run_eval.py +4 -0
.gitignore CHANGED
@@ -22,4 +22,4 @@ launch.json
22
  input_audios/
23
  syn_results/
24
  falcon-7b-instruct/
25
- flan-t5-large/
 
22
  input_audios/
23
  syn_results/
24
  falcon-7b-instruct/
25
+ flan-t5-xl/
run_eval.py CHANGED
@@ -83,6 +83,10 @@ def evaluate_system():
83
  rouge = evaluate.load("rouge")
84
  results = rouge.compute(predictions=predictions, references=references)
85
  print(f"ROUGE: {round(results['rougeL'], 2)}")
 
 
 
 
86
 
87
  if __name__ == "__main__":
88
  evaluate_system()
 
83
  rouge = evaluate.load("rouge")
84
  results = rouge.compute(predictions=predictions, references=references)
85
  print(f"ROUGE: {round(results['rougeL'], 2)}")
86
+
87
+ bertscore = evaluate.load("bertscore")
88
+ results = bertscore.compute(predictions=predictions, references=references, lang="en")
89
+ print(f"F1: {round(sum(results['f1'])/len(results['f1']))}")
90
 
91
  if __name__ == "__main__":
92
  evaluate_system()