INNER CODE UNIT · Python

_cls_compute_metrics

hooshvare/parsbert · scripts/utils.py:166

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)
    return _cls_compute_metrics(predictions, p.label_ids)


def rgr_compute_metrics(p: EvalPrediction) -> Dict:
    return _rgr_compute_metrics(p.predictions, p.label_ids)


def mode_compute_metrics(mode):

View source record →