How to plot ROC curve in Python

by

Here are two ways you may try, assuming your model is an sklearn predictor:

import sklearn.metrics as metrics
# calculate the fpr and tpr for all thresholds of the classification
probs = model.predict_proba(X_test)
preds = probs[:,1]
fpr, tpr, threshold = metrics.roc_curve(y_test, preds)
roc_auc = metrics.auc(fpr, tpr)

# method I: plt
import matplotlib.pyplot as plt
plt.title('Receiver Operating Characteristic')
plt.plot(fpr, tpr, 'b', label="AUC = %0.2f" % roc_auc)
plt.legend(loc="lower right")
plt.plot([0, 1], [0, 1],'r--')
plt.xlim([0, 1])
plt.ylim([0, 1])
plt.ylabel('True Positive Rate')
plt.xlabel('False Positive Rate')
plt.show()

# method II: ggplot
from ggplot import *
df = pd.DataFrame(dict(fpr = fpr, tpr = tpr))
ggplot(df, aes(x = 'fpr', y = 'tpr')) + geom_line() + geom_abline(linetype="dashed")

or try

ggplot(df, aes(x = 'fpr', ymin = 0, ymax = 'tpr')) + geom_line(aes(y = 'tpr')) + geom_area(alpha = 0.2) + ggtitle("ROC Curve w/ AUC = %s" % str(roc_auc))

More Related Contents:

  1. When to use cla(), clf() or close() for clearing a plot in matplotlib?
  2. Plotting time in Python with Matplotlib
  3. Matplotlib different size subplots
  4. Plotting in a non-blocking way with Matplotlib
  5. Hiding axis text in matplotlib plots
  6. How to add title to subplots in Matplotlib
  7. Format y axis as percent
  8. matplotlib savefig() plots different from show()
  9. Moving x-axis to the top of a plot in matplotlib
  10. matplotlib scatterplot with legend
  11. How can I rotate xticklabels in matplotlib so that the spacing between each xticklabel is equal?
  12. matplotlib advanced bar plot
  13. How to make a 4d plot using Python with matplotlib
  14. Pandas: Bar-Plot with two bars and two y-axis
  15. Plotting the data with scrollable x (time/horizontal) axis on Linux
  16. How can I connect scatter points in a 3D space?
  17. Pythonic way of detecting outliers in one dimensional observation data
  18. Putting text in top left corner of matplotlib plot
  19. Many plots in less time – python
  20. How do I tell Matplotlib to create a second (new) plot, then later plot on the old one?
  21. How do you directly overlay a scatter plot on top of a jpg image in matplotlib / Python?
  22. Center origin in matplotlib
  23. Repeating x axis labels for all facets using FacetGrid in seaborn
  24. How to make a 3D scatter plot in matplotlib
  25. Xticks by pandas plot, rename with the string
  26. Placing Custom Images in a Plot Window–as custom data markers or to annotate those markers
  27. When to use cla(), clf() or close() for clearing a plot
  28. How to plot a scatter plot with a legend label for each class
  29. Putting arrowheads on vectors in a 3d plot
  30. How to make a 3D scatter plot

Leave a Comment