在Python中创建阈值编码的ROC图

Question

R的ROCR软件包为ROC曲线绘图提供了选项,可以沿着曲线对代码和标签阈值进行着色:

我能用Python获得的最接近的东西就像

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])

这使

是否有提供与R标记(使用print.cutoffs.at)和颜色代码(使用colorize)阈值能力相当的功能的软件包？据推测,这些信息是thresholds由sklearn.metrics.roc_curve,但是我无法弄清楚如何使用它来为代码着色和标记图形.

Answer 1

看看这个要点:

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

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