使用python的混淆矩阵图

数据挖掘 Python 绘图 混淆矩阵
2022-02-18 12:13:40

我是一个可以使用 python 绘制以下图的函数:

在此处输入图像描述

1个回答

免责声明,

您可以使用Python 模块PyCM,以便将混淆矩阵及其评估参数保存在 html 报告中。

您可以使用以下命令:

from pycm import *
y_actu = [2, 0, 2, 2, 0, 1, 1, 2, 2, 0, 1, 2]
y_pred = [2, 0, 2, 2, 0, 2, 1, 2, 2, 0, 2, 2]
cm = ConfusionMatrix(y_actu, y_pred)
cm.save_html(filename,color=(100,50,250))

结果将如下所示: 在此处输入图像描述