Middle
Milyen metrikák alkalmasak egy bináris osztályozási modell minőségének értékelésére?
sobes.tech MI
Válasz az MI-től
A bináris osztályozási modell minőségének értékeléséhez a következő metrikák alkalmasak:
- Pontosság (Accuracy) — helyes előrejelzések aránya.
- Pozitív predikciók pontossága (Precision) — hány a pozitívnak jósolt eredmény valóban pozitív.
- Visszahívás (Recall) — hány a tényleges pozitív példák közül helyesen azonosított a modell.
- F1-score — a pontosság és a visszahívás harmonikus átlaga, hasznos kiegyensúlyozatlan osztályok esetén.
- ROC-AUC — a ROC-görbe alatti terület, mutatja az osztályok szétválasztásának minőségét.
Pythonban sklearn segítségével a F1-score kiszámításának példája:
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}")