Как создать матрицу путаницы из вывода индексов - PullRequest
Новая
       15

Как создать матрицу путаницы из вывода индексов

1 голос
/ 08 апреля 2020

Я работаю над моделированием изображений с использованием керас. Я получаю следующий вывод после подгонки модели: (классификация по нескольким меткам) 

len(test_generator):5

test_generator.classe:   

>[[0,2],[1,3],[0,1,2],[3,1,2,0],[2,1]]

Predictions as:

predicted_class_indices:  (predicted_class_indices=np.argmax(pred_prob,axis=1)
predicted_class_indices 
    >[3,3,0,1,2]

Prediction_genres:

>['Comedy', 'Comedy', 'Action', 'Adventure', 'Animation']

Labels_map:

>{'Action': 0, 'Adventure': 1, 'Animation': 2, 'Comedy': 3, 'Crime': 4}

Я пытаюсь создать матрицу путаницы, как показано ниже: * 

print(confusion_matrix(test_generator.classes,Prediction_genres))

Я получаю следующую ошибку :

ValueError: Вы, кажется, используете устаревшее представление данных с несколькими метками. Последовательности последовательностей больше не поддерживаются, вместо этого используйте двоичный массив или разреженную матрицу.

...