roc_auc_score()

Площадь под ROC-кривой.

Сигнатураroc_auc_score(y_true, y_score)

Оценивает способность модели ранжировать положительные объекты выше отрицательных. На вход подают вероятности (или оценки), а не метки. Значение 1.0 — идеал, 0.5 — случайное угадывание.

from sklearn.metrics import roc_auc_score

proba = clf.predict_proba(X_test)[:, 1]
print(roc_auc_score(y_test, proba))

← Все записи: scikit-learn