Запутанная матрица предсказанных меток - PullRequest
0 голосов
/ 17 апреля 2020

Я использую следующий фрагмент кода python для генерации нормализованной матрицы путаницы

import scikitplot as skplt 
skplt.metrics.plot_confusion_matrix(y_test, y_pred, normalize=True)

, и это дает мне график, показанный здесь . Тем не менее, я хотел повернуть текс по оси X (прогнозируемые значения, т. Е. Нейтральные, счастливые, грустные). Вместо горизонтальной линии я хочу, чтобы прогнозируемые значения на оси X были повернуты (как показано в здесь ) и сохранили истинные метки такими, какие они есть (горизонтальные). Как мы можем сделать это в Python

1 Ответ

0 голосов
/ 17 апреля 2020

Если вы хотите использовать scikitplot, я считаю, что в функции plot_confusion_matrix есть аргумент x_tick_rotation. Таким образом, вы можете повернуть xticks до 45 градусов следующим образом.

skplt.metrics.plot_confusion_matrix(y_test, y_pred, x_tick_rotation=45, normalize=True)
...