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

1 vote174 views1 answer
The name tf.summary.merge_all is deprecated. Please use tf.compat.v1.summary.merge_all
Python
0 vote268 views1 answer
How do you read from a dictionary in python
Python
3 votes223 views1 answer
How to make downloadable file in flask
Python Flask
2 votes172 views1 answer
Directory name python
Python
1 vote169 views1 answer
Não nulo pandas
Python
0 vote170 views1 answer
Check cudann
Python
5 votes168 views2 answers
Python plot two lines on same graph
Python
0 vote168 views1 answer
Remove dot from number python
Python
0 vote232 views1 answer
Replace by positions a string in a list with another string python
Python
6 votes202 views3 answers
Download image from url python 3
Python