如何正确绘制混淆矩阵

时间:2017-05-25 20:16:07

标签: python scikit-learn confusion-matrix

有谁知道如何绘制100个标签的混淆矩阵?我做了这些代码行,但我最终得到了一个混淆矩阵。代码适用于较少的类数,如5,但由于类的数量是100,没有明确的混淆矩阵。 enter image description here

y_pred = model.predict(X_test)
confmat = confusion_matrix(y_true=y_test, y_pred=y_pred)
print(confmat)
fig, ax = plt.subplots(figsize=(5, 5))
ax.matshow(confmat, cmap=plt.cm.Blues, alpha=0.3)
for i in range(confmat.shape[0]):
    for j in range(confmat.shape[1]):
        ax.text(x=j, y=i, s=confmat[i, j], va='center', ha='center')
plt.xlabel('predicted label')
plt.ylabel('true label')
plt.show()

0 个答案:

没有答案