绘制多标签混淆矩阵

2024-05-15 01:17:54 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在进行一个多标签文本分类任务,其中我有21个标签要预测。我的x\U测试&;你的测试是这样的:

y检验:array([1, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0], dtype=int64)

是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函数,但是我找不到一种方法来绘制混淆矩阵来分析我的模型的结果


Tags: 函数文本分类矩阵标签sklearnarraymatrix

热门问题