绘制多标签混淆矩阵

时间:2019-11-09 00:58:49

标签: python scikit-learn confusion-matrix

我正在执行“多标签文本分类”任务,其中有21个要预测的标签。我的x_test和y_test看起来像这样:

y_test: array([1, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0], dtype=int64)

y_pred: array([0.98, 0.01, 0.0, 0.74, 0.0, 0.0, 0.0, 0.0, 0.21, 0.0, 0.0, 0.0, 0.0, 0.0, 0.41, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], dtype=float32)

混淆矩阵:

    array([[[4564,  109],
            [  78, 4545]],

           [[7534,  120],
            [ 158, 1484]],

           [[8127,   79],
            [ 147,  943]],

           [[8376,   48],
            [ 108,  764]],

            ... 16 matrix 

           [[8948,   46],
            [  94,  208]]], dtype=int64)

尽管sklearn提供了multilabel_confusion_matrix功能,但是我找不到一种绘制混淆矩阵来分析模型结果的方法。

0 个答案:

没有答案