modefied mgsm

cb979e6c · Israel Abebe Azime · facf38ca · facf38ca
Commit cb979e6c authored May 15, 2024 by Israel Abebe Azime
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 9 deletions

lm_eval/tasks/afrimgsm/direct/utils.py lm_eval/tasks/afrimgsm/direct/utils.py +0 -9

No files found.
--- a/lm_eval/tasks/afrimgsm/direct/utils.py
+++ b/lm_eval/tasks/afrimgsm/direct/utils.py
-from sklearn.metrics import f1_score
-
-
-def weighted_f1_score(items):
-    unzipped_list = list(zip(*items))
-    golds = unzipped_list[0]
-    preds = unzipped_list[1]
-    fscore = f1_score(golds, preds, average="weighted")
-    return fscore
\ No newline at end of file