/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 vote145 views2 answers
Friends phoebe
Python
2 votes133 views4 answers
Update link python is python 3
Python
0 vote195 views1 answer
Write a python program to find fibonacci series upto a certain limit using while loop
Python
0 vote151 views1 answer
Find-squares-and-odd-numbers-in-the-given-list
Python
0 vote140 views1 answer
Convert numpy array to tfrecord and back
Python
-1 vote152 views1 answer
ValueError: cannot reshape array of size 98292 into shape (16382,1,28) site:stackoverflow.com
Python
0 vote165 views1 answer
Python array of tuples for loop
Python
3 votes131 views1 answer
Count how many duplicates python pandas
Python
24 votes146 views1 answer
Print type(x) in python
Python
0 vote144 views1 answer
Accuracy for each class
Python