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

4 votes311 views2 answers
Usage of thread in python
Python
22 votes300 views5 answers
If else python
Python
14 votes292 views3 answers
How to get user id discord.py
Python
34 votes311 views6 answers
How to open a file in python
Python
1 vote306 views1 answer
Kill and run process in windows python
Python
1 vote288 views2 answers
Remove 1st column pandas
Python
0 vote314 views1 answer
Python flatten one liner
Python
0 vote297 views1 answer
Install python 3.8 on wsl
Python
0 vote309 views1 answer
Request.body django
Python
1 vote297 views1 answer
Stdout.write python
Python