sklearn plot confusion matrix with labels

by

UPDATE:

In scikit-learn 0.22, there’s a new feature to plot the confusion matrix directly (which, however, is deprecated in 1.0 and will be removed in 1.2).

See the documentation: sklearn.metrics.plot_confusion_matrix

OLD ANSWER:

I think it’s worth mentioning the use of seaborn.heatmap here.

import seaborn as sns
import matplotlib.pyplot as plt     

ax= plt.subplot()
sns.heatmap(cm, annot=True, fmt="g", ax=ax);  #annot=True to annotate cells, ftm='g' to disable scientific notation

# labels, title and ticks
ax.set_xlabel('Predicted labels');ax.set_ylabel('True labels'); 
ax.set_title('Confusion Matrix'); 
ax.xaxis.set_ticklabels(['business', 'health']); ax.yaxis.set_ticklabels(['health', 'business']);

enter image description here

More Related Contents:

  1. How can I plot a confusion matrix? [duplicate]
  2. Why does the following code cant get value of n_fold = 1?
  3. matplotlib: can I create AxesSubplot objects, then add them to a Figure instance?
  4. Setting different color for each series in scatter plot on matplotlib
  5. Reverse Y-Axis in PyPlot
  6. Defining the midpoint of a colormap in matplotlib
  7. ImportError in importing from sklearn: cannot import name check_build
  8. How do I fit a sine curve to my data with pylab and numpy?
  9. How to highlight specific x-value ranges
  10. warning about too many open figures
  11. Efficiently create sparse pivot tables in pandas?
  12. How to increase/reduce the fontsize of x and y tick labels
  13. How to convert a .wav file to a spectrogram in python3
  14. How to remove or hide x-axis labels from a seaborn / matplotlib plot
  15. python matplotlib blit to axes or sides of the figure?
  16. Colorbar for matplotlib plot_surface command
  17. plot decision boundary matplotlib
  18. Large Dataset Polynomial Fitting Using Numpy
  19. scipy.curve_fit() returns multiple lines
  20. Get data points from Seaborn distplot
  21. Plot a 3D surface from {x,y,z}-scatter data
  22. Double donut chart in matplotlib
  23. How to decode color mapping in matplotlib’s Colormap?
  24. Stratified Sampling in Pandas
  25. Unable to “import matplotlib.pyplot as plt” in virtualenv
  26. pylab.hist(data, normed=1). Normalization seems to work incorrect
  27. Move spines in matplotlib 3d plot?
  28. Combine two Pyplot patches for legend
  29. Plotting a 3d cube, a sphere and a vector
  30. Matplotlib – Wrong overlapping when plotting two 3D surfaces on the same axes

Leave a Comment