我正在为多个标签数据绘制混淆矩阵，其中标签看起来像：
标签1：1，0，0，0
label2：0，1，0，0
label3：0，0，1，0
标签4：0，0，0，1
我可以使用以下代码成功分类。我只需要一些帮助来绘制混淆矩阵。
for i in range(4):
y_train= y[:,i]
print('Train subject %d, class %s' % (subject, cols[i]))
lr.fit(X_train[::sample,:],y_train[::sample])
pred[:,i] = lr.predict_proba(X_test)[:,1]
我使用以下代码打印混淆矩阵，但它始终返回2X2矩阵
prediction = lr.predict(X_train)
print(confusion_matrix(y_train, prediction))