Sobes.tech
Middle

Quais métricas são adequadas para avaliar a qualidade de um modelo de classificação binária?

sobes.tech IA

Resposta da IA

Para avaliar a qualidade de um modelo de classificação binária, as seguintes métricas são adequadas:

  • Acurácia (Accuracy) — proporção de previsões corretas.
  • Precisão (Precision) — quantos dos positivos previstos são realmente positivos.
  • Recall (Sensibilidade) — quantos de todos os exemplos positivos o modelo identificou corretamente.
  • F1-score — média harmônica de precisão e recall, útil em classes desequilibradas.
  • ROC-AUC — área sob a curva ROC, indica a qualidade da separação das classes.

Exemplo de cálculo do F1-score em Python usando 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}")