INNER CODE UNIT · Python

mse_and_mae_and_r2

hooshvare/parsbert · scripts/utils.py:158

def mse_and_mae_and_r2(predictions, labels):
    return {
        "mse": mean_squared_error(labels, predictions),
        "mae": mean_absolute_error(labels, predictions),
        "r2": r2_score(labels, predictions),
    }


def _cls_compute_metrics(predictions, labels):
    return acc_and_f1(predictions, labels)


def _rgr_compute_metrics(predictions, labels):
    return acc_and_f1(predictions, labels)


def cls_compute_metrics(p: EvalPrediction) -> Dict:
    predictions = np.argmax(p.predictions, axis=1)

View source record →