Plot Confusion Matrix with custom x and y axis in Python?

Question

I have confusion matrix in form of an array like

[[ 265, 1522],
 [  44,  146]]

I am doing something like this to plot

import seaborn as sns
import matplotlib.pyplot as plt
cm = confusion_matrix(yTest, yPredicted)
sns.heatmap(cm, annot=True,fmt="d", cmap='YlGnBu', linecolor='black', linewidths=1)
plt.title("Confusion Matrix")
plt.title("Confusion Matrix")
plt.xlabel("Actual")
plt.ylabel("Predicted")

and

But How can i change x and y axsis and get graph like that

Stef · Accepted Answer · 2021-01-05 09:45:29Z

1

Transpose the matrix, swap the axis labels and reverse the axis directions by setting the limits:

sns.heatmap(cm.T, annot=True,fmt="d", cmap='YlGnBu', linecolor='black', linewidths=1)
plt.title("Confusion Matrix")
plt.ylabel("Actual")
plt.xlabel("Predicted")
plt.ylim(0,2)
plt.xlim(2,0)

You can rotate y axis labels as shown in your example like that:

plt.ylabel('\n'.join(list('Actual')), rotation=0, va='center')
plt.yticks(rotation=0)

edited Jan 5, 2021 at 9:45

answered Jan 5, 2021 at 9:31

Stef

30.9k3 gold badges34 silver badges60 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Plot Confusion Matrix with custom x and y axis in Python?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related