ValueError: неизвестный тип метки для классификационного отчета - PullRequest
Новая
       57

ValueError: неизвестный тип метки для классификационного отчета

0 голосов
/ 30 июня 2019

Я пытаюсь оценить модель для мультиклассовой классификации с помощью модуляification_report пакета sklean.

Размеры y_pred: (1000,36) Размеры y_test: (1000,36)

Я попытался вызвать классификационный отчет по двум массивам, т. Е. Y_test и y_pred 

def display_results(y_test,y_pred,column_name=labels):
    print(classification_report(y_test,y_pred,target_names=labels))

С этим кодом я получаю: 

ValueError: Unknown label type: (array([[1, 0, 0, ..., 0, 0, 0],
       [1, 0, 0, ..., 0, 0, 0],
       [1, 0, 0, ..., 1, 1, 0],
       ...,
       [1, 0, 0, ..., 0, 0, 0],
       [0, 0, 0, ..., 0, 0, 0],
       [1, 0, 0, ..., 0, 0, 0]]), array([[1, 0, 0, ..., 0, 0, 0],
       [1, 0, 0, ..., 0, 0, 0],
       [1, 0, 0, ..., 0, 0, 0],
       ...,
       [1, 0, 0, ..., 0, 0, 0],
       [0, 0, 0, ..., 0, 0, 0],
       [1, 0, 0, ..., 0, 0, 0]]))

Я ожидал получить Precision, Recall, F1 и общее среднее значение для всех столбцов на основе меток, переданных в функцию.

...