/Python/Compute confusion matrix using python

Compute confusion matrix using python

import numpy as np

currentDataClass = [1, 3, 3, 2, 5, 5, 3, 2, 1, 4, 3, 2, 1, 1, 2]
predictedClass = [1, 2, 3, 4, 2, 3, 3, 2, 1, 2, 3, 1, 5, 1, 1]

def comp_confmat(actual, predicted):

    classes = np.unique(actual) # extract the different classes
    matrix = np.zeros((len(classes), len(classes))) # initialize the confusion matrix with zeros

    for i in range(len(classes)):
        for j in range(len(classes)):

            matrix[i, j] = np.sum((actual == classes[i]) & (predicted == classes[j]))

    return matrix

comp_confmat(currentDataClass, predictedClass)

array([[3., 0., 0., 0., 1.],
       [2., 1., 0., 1., 0.],
       [0., 1., 3., 0., 0.],
       [0., 1., 0., 0., 0.],
       [0., 1., 1., 0., 0.]])

Source: stackoverflow.com

Similar codes

0 vote178 views1 answer
Python string after character
Python
0 vote174 views1 answer
Remove trailing white space python string
Python
0 vote160 views1 answer
Python - match two df on a variable with different name
Python
2 votes163 views2 answers
Change the color of the button on hovering tkinter
Python
0 vote173 views1 answer
Urllib.error.HTTPError: HTTP Error 403: Forbidden
Python
0 vote154 views1 answer
Django not migrating
Python
14 votes167 views2 answers
How to get device name using pythno
Python
89 votes150 views6 answers
How to use for in python
Python
13 votes204 views4 answers
How to use if else in lambda python
Python
33 votes142 views2 answers
Find a largest number in function list python
Python