多类别,多标签分类问题的混淆矩阵

时间:2019-11-22 15:56:25

标签: python scikit-learn confusion-matrix

我试图为Python找到一个函数,该函数针对多类和多标签分类问题计算了混淆矩阵,因此预测和目标如下:

y_true = np.array([[1, 0, 1],
                   [0, 1, 0]])
y_pred = np.array([[1, 0, 0],
                   [0, 1, 1]])

我在scykit-learn library a function中发现,它为每个类以“一对多”的方式绘制了多个二进制混淆矩阵:

[[[1, 0],
    [0, 1]],

   [[1, 0],
    [0, 1]],

   [[0, 1],
    [1, 0]]]

相反,我想像一个简单的多类问题中那样使用一个混淆矩阵,例如:https://i.stack.imgur.com/WDNzF.png 到目前为止,我还没有找到一个可以让您这样做的python库,我是否缺少一些无法做到的东西?

0 个答案:

没有答案