Update evaluation scripts

Waste-Wood · Mar 29, 2022 · 13c9895 · 13c9895
1 parent e558f18
commit 13c9895
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/evaluation_metrics_causal_reasoning.py b/evaluation_metrics_causal_reasoning.py
@@ -26,6 +26,7 @@ def evaluation_metrics(gold, predictions):
     return accuracy
 
 
+
 def main():
     prediction_file = sys.argv[1]
     gold_file = sys.argv[2]

diff --git a/evalution_metrics_conceptual_explanation_generation.py b/evalution_metrics_conceptual_explanation_generation.py
@@ -27,7 +27,7 @@ def evaluation_bleu(gold, predictions):
 
 
 def evaluation_rouge(golds, predictions):
-    rougel = 0
+    rouge_l = 0
     rouge = Rouge()
 
     for key in predictions:
@@ -39,11 +39,11 @@ def evaluation_rouge(golds, predictions):
 
         try:
             scores = rouge.get_scores(prediction, gold)
-            rougel += scores[0]['rouge-l']['r']
+            rouge_l += scores[0]['rouge-l']['r']
         except:
             continue
 
-    avg_rougel = rougel / len(golds)
+    avg_rougel = rouge_l / len(golds)
     return avg_rougel