我正在进行一个多标签文本分类任务,其中我有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
函数,但是我找不到一种方法来绘制混淆矩阵来分析我的模型的结果
目前没有回答
相关问题 更多 >
编程相关推荐