/Python/Print labels on confusion_matrix

Print labels on confusion_matrix

unique_label = np.unique([y_true, y_pred])
cmtx = pd.DataFrame(
    confusion_matrix(y_true, y_pred, labels=unique_label), 
    index=['true:{:}'.format(x) for x in unique_label], 
    columns=['pred:{:}'.format(x) for x in unique_label]
)
print(cmtx)
# Output:
#           pred:no  pred:yes
# true:no         3         0
# true:yes        2         1

Source: stackoverflow.com

Print labels on confusion_matrix

import pandas as pd
cmtx = pd.DataFrame(
    confusion_matrix(y_true, y_pred, labels=['yes', 'no']), 
    index=['true:yes', 'true:no'], 
    columns=['pred:yes', 'pred:no']
)
print(cmtx)
# Output:
#           pred:yes  pred:no
# true:yes         1        2
# true:no          0        3

Source: stackoverflow.com

Similar codes

103 votes345 views6 answers
Rename column pandas
Python
0 vote331 views1 answer
Python pipe where
Python
5 votes352 views3 answers
Pandas convert string to float
Python
0 vote356 views1 answer
Python string: escaping characters
Python
1 vote356 views1 answer
Django sort descending
Python Django
19 votes382 views3 answers
Map python 3
Python
3 votes318 views1 answer
Delete a key value pair from a dictionary in python
Python
1 vote327 views1 answer
Keras declare functional model
Python
1 vote348 views2 answers
Creating lambda expressions in comprehension list
Python
0 vote381 views1 answer
How add a favicon to django
Python