python - forest - roc_auc_score scikit learn

Crear un diagrama ROC con código de umbral en Python

Question

El paquete ROCR de R proporciona opciones para el trazado de la curva ROC que codificará el color y etiquetará los valores de umbral a lo largo de la curva:

Lo más cercano que puedo obtener con Python es algo así como

from sklearn.metrics import roc_curve
fpr, tpr, thresholds = roc_curve(qualityTrain.PoorCare, qualityTrain.Pred1)
plt.plot(fpr, tpr, label=''ROC curve'', color=''b'')
plt.axes().set_aspect(''equal'')
plt.xlim([-0.05, 1.05])
plt.ylim([-0.05, 1.05])

lo que da

¿Existen paquetes que brinden una funcionalidad equivalente a la capacidad de R para etiquetar (utilizando print.cutoffs.at ) y el código de color (usar colorize ) umbrales? Presumiblemente esta información está en thresholds , devuelta por sklearn.metrics.roc_curve , pero no puedo encontrar la forma de usarla para colorear el código y etiquetar la figura.

Answer 1

Mira esta esencia:

https://gist.github.com/podshumok/c1d1c9394335d86255b8

roc_data = sklearn.metrics.roc_curve(...)
plot_roc(*roc_data, label_every=5)