我正在执行“多标签文本分类”任务,其中有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
功能,但是我找不到一种绘制混淆矩阵来分析模型结果的方法。