diff --git a/examples/plot_roc.py b/examples/plot_roc.py index d2d492f5a3abc3e236d760ee7a9e13e7cacfd487..884aaa84cd6c2af5b44ffdb07cc6d2dd910daaf9 100644 --- a/examples/plot_roc.py +++ b/examples/plot_roc.py @@ -73,7 +73,8 @@ for i in range(n_classes): fpr["micro"], tpr["micro"], _ = roc_curve(y_test.ravel(), y_score.ravel()) roc_auc["micro"] = auc(fpr["micro"], tpr["micro"]) -# Plot of one ROC curve +# Plot of a ROC curve for a specific class +plt.figure() plt.plot(fpr[2], tpr[2], label='ROC curve (area = %0.2f)' % roc_auc[2]) plt.plot([0, 1], [0, 1], 'k--') plt.xlim([0.0, 1.0]) @@ -85,6 +86,7 @@ plt.legend(loc="lower right") plt.show() # Plot ROC curve +plt.figure() plt.plot(fpr["micro"], tpr["micro"], label='micro-average ROC curve (area = {0:0.2f})' ''.format(roc_auc["micro"]))