Accuracy for each class

from sklearn.metrics import confusion_matrix
import numpy as np

y_true = [0, 1, 2, 2, 2]
y_pred = [0, 0, 2, 2, 1]
target_names = ['class 0', 'class 1', 'class 2']

#Get the confusion matrix
cm = confusion_matrix(y_true, y_pred)
#array([[1, 0, 0],
#   [1, 0, 0],
#   [0, 1, 2]])

#Now the normalize the diagonal entries
cm = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]
#array([[1.        , 0.        , 0.        ],
#      [1.        , 0.        , 0.        ],
#      [0.        , 0.33333333, 0.66666667]])

#The diagonal entries are the accuracies of each class
cm.diagonal()
#array([1.        , 0.        , 0.66666667])

Source: stackoverflow.com

Similar codes

7 votes230 views1 answer
If else python in single line
Python
7 votes206 views1 answer
Coinflip
Python
-2 votes209 views6 answers
List comprehension in python
Python
0 vote197 views1 answer
Two input string sum in django
Python
0 vote258 views2 answers
Url namespaces for django rest router urls
Python Django
0 vote238 views1 answer
?: (corsheaders.E013) Origin '.' in CORS_ORIGIN_WHITELIST is missing scheme or netloc HINT: Add a scheme (e.g. https://) or netloc (e.g. example.com).
Python
2 votes201 views1 answer
Matplotlib bar label
Python
0 vote249 views3 answers
Django add user to group
Python Django
0 vote219 views2 answers
How to find common characters in two strings in python
Python
10 votes290 views2 answers
Set axis labels python
Python