Я использую следующий код для построения матрицы путаницы:
labels = test_y.unique()
predictions = chosen_clf.predict(X=test_x)
conf_matrix = confusion_matrix(y_true=test_y, y_pred=predictions, labels=labels)
conf_matrix = pd.DataFrame(conf_matrix, index=labels, columns=labels)
plt.figure()
sn.heatmap(conf_matrix, annot=True)
plt.savefig(r'confusion_matrix.png')
Однако при получении графика возникают некоторые проблемы: - метки обрезаются - сетка недостаточно широка, что приводит кс нечитаемыми значениями, например, 2e + 02 в положении (1,1).
Как это исправить?
