ValueError：混淆矩阵不支持multilabel-indicator

Question

在我的二进制分类问题中尝试使用混淆矩阵时遇到此错误。 Y和Yhat都是numpy数组。我尝试使用.argmax作为建议的解决方案-我不再收到错误，但输出不是我所知道的混淆矩阵。

Accuracy: 0.9982449999999999
Accuracy: 0.9983374013937532

shape of y =  (1, 200000)
shape of yhat =  (1, 200000)

错误

 ValueError                                Traceback (most recent call last) <ipython-input-13-ebb660b4585a> in <module>()
     12 print("shape of yhat = ", yhat.shape)
     13 
---> 14 cm = confusion_matrix(y,yhat)
     15 
     16 print("confusion matrix = ", cm)

/anaconda3/lib/python3.6/site-packages/sklearn/metrics/classification.py in confusion_matrix(y_true, y_pred, labels, sample_weight)
    250     y_type, y_true, y_pred = _check_targets(y_true, y_pred)
    251     if y_type not in ("binary", "multiclass"):
--> 252         raise ValueError("%s is not supported" % y_type)
    253 
    254     if labels is None:

ValueError: multilabel-indicator is not supported

Answer 1

按如下所示添加代码

cm = confusion_matrix(y.argmax(axis=1),yhat.argmax(axis=1))

在之前问过here