Spaces:
Runtime error
Runtime error
lhzstar
commited on
Commit
•
b190683
1
Parent(s):
7fba753
new commits
Browse files- .gitignore +1 -1
- 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-
|
|
|
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()
|