Sobes.tech
Middle

Ce metrici sunt potrivite pentru evaluarea calității unui model de clasificare binară?

sobes.tech AI

Răspuns de la AI

Pentru evaluarea calității unui model de clasificare binară, următoarele metrici sunt potrivite:

  • Acuratețea (Accuracy) — proporția de predicții corecte.
  • Precizia (Precision) — câte dintre predicțiile pozitive sunt cu adevărat pozitive.
  • Recall (Sensibilitate) — câte dintre toate exemplele pozitive au fost corect identificate.
  • F1-score — media armonică a preciziei și recall-ului, util în cazurile cu clase dezechilibrate.
  • ROC-AUC — aria de sub curba ROC, indică calitatea separării claselor.

Exemplu de calcul al F1-score în Python cu sklearn:

from sklearn.metrics import f1_score

y_true = [0, 1, 1, 0, 1]
y_pred = [0, 1, 0, 0, 1]

score = f1_score(y_true, y_pred)
print(f"F1-score: {score}")