Мультиклассовая запутанная матрица с Sklearn - PullRequest
Новая
       72

Мультиклассовая запутанная матрица с Sklearn

0 голосов
/ 21 декабря 2018

Я работаю с мультиклассовым выводом нескольких меток из моего классификатора.Общее количество классов составляет 14, и экземпляры могут иметь несколько связанных классов.Например: 

y_true = np.array([[0,0,1], [1,1,0],[0,1,0])
y_pred = np.array([[0,0,1], [1,0,1],[1,0,0])

То, как я сейчас делаю свою матрицу путаницы: 

matrix = confusion_matrix(y_true.argmax(axis=1), y_pred.argmax(axis=1))
print(matrix)

, которая дает вывод типа: 

[[ 79   0   0   0  66   0   0 151   1   8   0   0   0   0]
 [  4   0   0   0  11   0   0  27   0   0   0   0   0   0]
 [ 14   0   0   0  21   0   0  47   0   1   0   0   0   0]
 [  1   0   0   0   4   0   0  25   0   0   0   0   0   0]
 [ 18   0   0   0  50   0   0  63   0   3   0   0   0   0]
 [  4   0   0   0   3   0   0  19   0   0   0   0   0   0]
 [  2   0   0   0   3   0   0  11   0   2   0   0   0   0]
 [ 22   0   0   0  20   0   0 138   1   5   0   0   0   0]
 [ 12   0   0   0   9   0   0  38   0   1   0   0   0   0]
 [ 10   0   0   0   3   0   0  40   0   4   0   0   0   0]
 [  3   0   0   0   3   0   0  14   0   3   0   0   0   0]
 [  0   0   0   0   2   0   0   3   0   0   0   0   0   0]
 [  2   0   0   0  11   0   0  32   0   0   0   0   0   0]
 [  0   0   0   0   0   0   0   3   0   0   0   0   0   7]]

Теперь яЯ не уверен, что матрица путаницы из sklearn способна обрабатывать мультиклассовые мультиклассовые данные.Может ли кто-нибудь помочь мне с этим?

Ответы [ 2 ]

0 голосов
/ 27 мая 2019

Теперь вы можете использовать (версия 0.21) sklearn.metrics.multilabel_confusion_matrix

https://scikit -learn.org / stable / modules / generate / sklearn.metrics.multilabel_confusion_matrix.html

Мы пытаемся предсказать две метки для каждого примера 

import sklearn.metrics as skm
y_true = np.array([
                [0,0], [0,1], [1,1], [0,1], [0,1], [1,1]
              ])
 y_pred = np.array([
                [1,1], [0,1], [0,1], [1,0], [0,1], [1,1] 
              ])

 cm = skm.multilabel_confusion_matrix(y_true, y_pred)
 print(cm)
 print( skm.classification_report(y_true,y_pred))

Матрица путаницы для меток: 

[[[2 2]
  [1 1]]

 [[0 1]
  [1 4]]]

Отчет о классификации: 

              precision    recall  f1-score   support

         0       0.33      0.50      0.40         2
         1       0.80      0.80      0.80         5

micro avg        0.62      0.71      0.67         7
macro avg        0.57      0.65      0.60         7
weighted avg     0.67      0.71      0.69         7
samples avg      0.67      0.58      0.61         7
0 голосов
/ 21 декабря 2018

Что вам нужно сделать, это сгенерировать несколько двоичных матриц путаницы (поскольку, по сути, у вас есть несколько двоичных меток)

Что-то вроде: 

import numpy as np
from sklearn.metrics import confusion_matrix

y_true = np.array([[0,0,1], [1,1,0],[0,1,0]])
y_pred = np.array([[0,0,1], [1,0,1],[1,0,0]])

labels = ["A", "B", "C"]

conf_mat_dict={}

for label_col in range(len(labels)):
    y_true_label = y_true[:, label_col]
    y_pred_label = y_pred[:, label_col]
    conf_mat_dict[labels[label_col]] = confusion_matrix(y_pred=y_pred_label, y_true=y_true_label)


for label, matrix in conf_mat_dict.items():
    print("Confusion matrix for label {}:".format(label))
    print(matrix)
...