在我的多标签分类(6个标签)中,预测结果和真实标签如下:
y_pred
array([[0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0.],
...,
[0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0.]], dtype=float32)
y_test
array([[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
...,
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0]])
我在此post中使用
from sklearn.metrics import confusion_matrix
confusion_matrix(y_test.argmax(axis=1), y_pred.argmax(axis=1) )
我困惑的矩阵丢失了一个标签,如下所示,只有5个标签:
array([[31635, 51, 2, 22, 18],
[ 88, 2, 0, 1, 0],
[ 7, 0, 0, 0, 0],
[ 76, 2, 0, 0, 1],
[ 9, 0, 0, 0, 1]])
希望有人知道它发生的原因。干杯。