shounakpaul95 commited on
Commit
433f9e3
1 Parent(s): 4bed9ae

Update eval_utils.py

Browse files
Files changed (1) hide show
  1. eval_utils.py +1 -1
eval_utils.py CHANGED
@@ -89,7 +89,7 @@ def evaluate_cjpe(gold_data, pred_data):
89
  }
90
  print("Explanability for ILDC Expert:", explanation_result)
91
  #return {**prediction_result, **explanation_result}
92
- return {"mF1": f1*100, "ROUGE-L": rouge_score*100, "BLEU": bleu_score*100}
93
 
94
  def span2bio(txt, roles):
95
  roles = sorted(roles, key = lambda x:x['start'])
 
89
  }
90
  print("Explanability for ILDC Expert:", explanation_result)
91
  #return {**prediction_result, **explanation_result}
92
+ return {"cjpe-prediction": {"mF1": f1*100}, "cjpe-explanation": {"ROUGE-L": rouge_score*100, "BLEU": bleu_score*100}}
93
 
94
  def span2bio(txt, roles):
95
  roles = sorted(roles, key = lambda x:x['start'])