如何绘制多类混淆矩阵？

Question

我正在尝试为我的主题分类绘制一个混淆矩阵。然而，当我绘制它时，我看不清楚任何东西。我得到了潦草的X和Y轴没有任何正确的绘图。如何以更好的方式绘制它以查看我的文档如何被分类？我有22465份文件和88个主题。

这是我的代码：

from pandas_confusion import ConfusionMatrix
import matplotlib.pyplot as plt
model = LogisticRegression()
model = model.fit(matrix_tmp, label_tmp)
pred = model.predict(matrix_tmp_test)

print(metrics.accuracy_score(label_tmp_test,pred))
print(metrics.classification_report(label_tmp_test,pred))

cm=ConfusionMatrix(label_tmp_test,pred)
print(cm)
cm.plot()
plt.show()

这是我收到的情节。

Answer 1

免责声明， 你好 PyCM是一个python模块，可以帮助您通过不同类型的报告（例如html报告）显示多类混淆矩阵，并对其进行绘制。

有一个简单的代码可以保存html混淆矩阵报告。

cm.save_html("file_name",color=(R,G,B))