/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

1 vote205 views1 answer
Tkinter radio buttons
Python
0 vote200 views1 answer
Get only every 2 rows pandas
Python
0 vote213 views1 answer
Matplotlib scatter increase marker size
Python
1 vote201 views1 answer
How to declare private attribute in python
Python
30 votes212 views2 answers
How do i limit decimals to only two decimals in python
Python
12 votes201 views3 answers
Pandas xlsx to dataframe
Python
31 votes194 views6 answers
Round off python
Python
11 votes208 views1 answer
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xe2 in position 10: invalid continuation byte
Python
1 vote192 views1 answer
Renpy scene vs show
Python
10 votes203 views2 answers
Create dict from json file python
Python