shounakpaul95 commited on
Commit
25db2bc
1 Parent(s): 82fce5e

Update eval_utils.py

Browse files
Files changed (1) hide show
  1. eval_utils.py +2 -2
eval_utils.py CHANGED
@@ -238,8 +238,8 @@ def evaluate_pcr(gold_data, pred_data):
238
 
239
  print(f"Micro-F1@{k} on IL-PCR test set:", f1)
240
 
241
- max_f1 = max(f1)
242
- index_max = f1.index(max_f1) + 1
243
  return f"{max_f1:.2f}@{index_max}"
244
 
245
 
 
238
 
239
  print(f"Micro-F1@{k} on IL-PCR test set:", f1)
240
 
241
+ max_f1 = max(f1_scores)
242
+ index_max = f1_scores.index(max_f1) + 1
243
  return f"{max_f1:.2f}@{index_max}"
244
 
245