Home > Enterprise >  How to sort a confusion matrix by the diagonal value
How to sort a confusion matrix by the diagonal value

Time:07-02

I have this confusion matrix:

import pandas as pd
import seaborn as sn
import matplotlib.pyplot as plt

data = {'y_Actual':    [3, 3, 1, 1, 0, 1, 2, 3, 1, 1, 1, 0, 2, 4, 3],
        'y_Predicted': [1, 2, 2, 1, 0, 1, 3, 0, 1, 0, 0, 0, 3, 4, 2]
        }

df = pd.DataFrame(data, columns=['y_Actual','y_Predicted'])
confusion_matrix = pd.crosstab(df['y_Actual'], df['y_Predicted'],
                               rownames=['Actual'], colnames=['Predicted'])

sn.heatmap(confusion_matrix, annot=True)
plt.show()

enter image description here

Is it possible to sort this confusion matrix by diagonal value in descending order ?

CodePudding user response:

You can use enter image description here

  • Related