textfier.utils.metrics

Metrics-based utilities for computing important information.

textfier.utils.metrics.compute_metrics(preds: transformers.trainer_utils.PredictionOutput)

Computes a set of metrics (accuracy, f1-score, precision and recall).

Parameters

preds – Predictions and potential metrics.

Returns

Set of metrics.

Return type

(Dict[str, Any])