/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

14 votes103 views2 answers
Select DF columns python
Python
25 votes106 views3 answers
Pyperclip
Python
0 vote114 views1 answer
Python consecutive numbers difference between
Python
0 vote96 views1 answer
Python if condition assignment in one line
Python
22 votes101 views1 answer
Python isset key
Python
32 votes99 views3 answers
Python move files
Python
0 vote70 views1 answer
Off-by-one error in python
Python
0 vote110 views2 answers
How to get value from txtbox in flask
Python
0 vote110 views1 answer
Python json random number generator
Python
5 votes122 views1 answer
UnicodeDecodeError: 'charmap' codec can't decode byte 0x9e in position 3359: character maps to <undefined>
Python