Вывод обрезанного графика (тепловая карта путаницы) - PullRequest
0 голосов
/ 01 марта 2020

У меня вывод матрицы путаницы выглядит обрезанным (см. Рисунок)

cm

Ну, это должно быть примерно так

cm2

Я не знаю, что не так с моим кодом ... вот мой код:

labels = [1.0, 0.0]
cm = confusion_matrix(y_act, y_pred, labels)     

ax= plt.subplot()
sns.heatmap(cm, annot=True, ax = ax); #annot=True to annotate cells

# labels, title and ticks
ax.set_xlabel('Predicted labels');ax.set_ylabel('True labels'); 
ax.set_title('Confusion Matrix'); 
ax.xaxis.set_ticklabels(labels); ax.yaxis.set_ticklabels(labels);

Может ли кто-нибудь помочь мне решить эту проблему

...