python - 是否可以绘制包含 90 个类的混淆矩阵？

Question

我想为我的分类模型绘制混淆矩阵。它有大约 20000 个文档需要分类到 90 个类。我收到的混淆矩阵很大。我想绘制这个，但我似乎到处都只能找到二进制分类图。是否可以绘制这个多类混淆矩阵？我尝试了一些方法，但显示不清晰。

这是我的混淆矩阵的样子:

[[3919  344    0 ...,    0    0    1]
 [ 267 2739    0 ...,    0    0    0]
 [   1    6   17 ...,    0    0    0]
 ..., 
 [   4    1    0 ...,    6    0    0]
 [   0    2    0 ...,    0    0    0]
 [   6    1    0 ...,    0    0   15]]

Answer 1

这是一些使用 matplotlib 的示例代码 (编辑:添加网格并关闭插值)

import numpy as np
import matplotlib.pyplot as plt

confmat=np.random.rand(90,90)
ticks=np.linspace(0, 89,num=90)
plt.imshow(confmat, interpolation='none')
plt.colorbar()
plt.xticks(ticks,fontsize=6)
plt.yticks(ticks,fontsize=6)
plt.grid(True)
plt.show()