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

5 votes142 views3 answers
How to add button in tkinter
Python
0 vote173 views1 answer
Get random bright hex color python
Python
5 votes182 views1 answer
Python flask query params
Python Flask
0 vote161 views1 answer
Index.py:14: RuntimeWarning: invalid value encountered in true_divide return np.dot(user, user2) / (norm(user) * norm(user2))
Python
4 votes149 views5 answers
Module 'datetime' has no attribute 'now' django
Python
-1 vote144 views1 answer
Loading .dat file in python
Python
2 votes160 views2 answers
Python greatest common divisor
Python
0 vote144 views2 answers
Insert character into string at position x python
Python
0 vote177 views1 answer
Pls help i need tkintwr help plspslspslspsl help tkinter
Python
0 vote152 views1 answer
Python loop chrome
Python