【深度学习实战（52）】混淆矩阵计算

class ConfusionMatrix(object):

    def __init__(self, class_num: int):
        self.matrix = numpy.zeros((class_num, class_num))
        self.class_num = class_num
        self.pred=[]
        self.true=[]

    def update(self, pred, label):
        # p代表Predicted label、t代表True label
        for p, t in zip(pred, label):
            self.matrix[p, t] += 1
            self.pred.append(p)
            self.true.append(t)

    def acc(self):
        acc = 0
        for i in range(self.class_num):
            acc += self.matrix[i, i]
        acc = acc / numpy.sum(self.matrix)

        return acc
    def report(self):
        report=classification_report(self.true,self.pred,digits=4)
        return report