utils.py 165 Bytes
Newer Older
1
from lm_eval.utils import weighted_f1_score
2
3
4


def doc_to_target(doc):
Lintang Sutawika's avatar
Lintang Sutawika committed
5
    replacements = {0: "True", 1: "Neither", 2: "False"}
6
    return replacements[doc["label"]]