/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

0 vote210 views1 answer
From flask_paginate import get_page_parameter
Python
60 votes228 views6 answers
Python current date
Python
0 vote218 views1 answer
Latex maths to python parser
Python
0 vote259 views1 answer
How to select the shortest item in a python list
Python
0 vote229 views1 answer
Append path to sys jupyter notebook
Python
0 vote213 views1 answer
Python emoji convert
Python
1 vote222 views2 answers
Tensor vs numpy array
Python
14 votes300 views2 answers
How to delete the last item in a list python
Python
1 vote245 views1 answer
Import CSV Files into R Using read_csv() method
Python
11 votes238 views1 answer
Pandas xlsx to csv
Python