INNER CODE UNIT · Python

Split

hooshvare/parsbert · scripts/utils.py:138

class Split(Enum):
    train = "train"
    dev = "dev"
    test = "test"


def simple_accuracy(predictions, labels):
    return (predictions == labels).mean()


def acc_and_f1(predictions, labels):
    acc = simple_accuracy(predictions, labels)
    f1 = f1_score(y_true=labels, y_pred=predictions, average='weighted')
    return {
        "acc": acc,
        "f1": f1,
        "acc_and_f1": (acc + f1) / 2,
    }

View source record →