pandas - matplotlib 中带有标签的混淆矩阵未正确组织

Question

我正在使用以下代码来绘制混淆矩阵：

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')

然而，我得到的情节有一些问题： - 标签被剪切 - 网格不够宽导致不可读的值，例如 (1,1) 位置的 2e+02。

我怎样才能解决这个问题？

标签： pandasmatplotlibseabornconfusion-matrix

Answer 1

尝试添加：

plt.figure(figsize=(20,20))

行前：

sn.heatmap()