Fixing Confusion Matrix plot lines

user_14167873 0
Viewed 1890

I am trying to plot a confusion matrix as shown below

cm  = confusion_matrix(testY.argmax(axis=1), predictions.argmax(axis=1))

disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=lb.classes_)
disp = disp.plot(include_values=True, cmap='viridis', ax=None, xticks_rotation='horizontal')

plt.show()

The result:

Confusion Matrix I get

As you can see, it's showing the axes of the boxes instead of outlining the boxes. I can't see the numbers outside the yellow boxes, because of the axes. I am not good with plots. So I can't find out what I need to change.

What I expect: Expected Matrix

FOUND SOLUTION

plt.tick_params(axis=u'both', which=u'both',length=0)
plt.grid(b=None)
confusion-matrix scikit-learn python
4 Answers

Turn the grid off

E.g.,

import matplotlib.pyplot as plt
fig, _ = plt.subplots(nrows=1, figsize=(10,10))
ax = plt.subplot(1, 1, 1)
ax.grid(False)

...

disp = ConfusionMatrixDisplay(...)
_ = disp.plot(..., ax=ax, ...)
user_68270 
cm  = confusion_matrix(testY.argmax(axis=1), predictions.argmax(axis=1))

disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=lb.classes_)
disp = disp.plot(include_values=True, cmap='viridis', ax=None, xticks_rotation='horizontal')
plt.grid(False)
plt.show()
user_69382340

Change your cmap parameter in plot() function. It stands for colour-mapping your integer values with colors.

Check

https://matplotlib.org/3.1.0/tutorials/colors/colormaps.html

for more details.

As the answer

cm  = confusion_matrix(testY.argmax(axis=1), predictions.argmax(axis=1))

disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=lb.classes_)
disp = disp.plot(include_values=True, cmap='Blues', ax=None, xticks_rotation='horizontal')

plt.show()
user_56644830

The graph which you are showing as example is by sns plot. You can use sns heatmap to plot your matrix.

import seaborn as sns
categories = lb.classes_
sns.heatmap(cm, annot=True,categories =categories, cmap='Blues')
user_64850160
Related