Spaces:
Running
Running
shounakpaul95
commited on
Commit
•
25db2bc
1
Parent(s):
82fce5e
Update eval_utils.py
Browse files- 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(
|
242 |
-
index_max =
|
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 |
|