python多分类混淆矩阵代码_如何计算Scikit中多类分类的混淆矩阵？

我有一个多类分类任务。当我运行基于scikit example的脚本时，如下所示：classifier = OneVsRestClassifier(GradientBoostingClassifier(n_estimators=70, max_depth=3, learning_rate=.02))

y_pred = classifier.fit(X_train, y_train).predict(X_test)

cnf_matrix = confusion_matrix(y_test, y_pred)

我得到这个错误：File "C:\ProgramData\Anaconda2\lib\site-packages\sklearn\metrics\classification.py", line 242, in confusion_matrix

raise ValueError("%s is not supported" % y_type)

ValueError: multilabel-indicator is not supported

我试着把labels=classifier.classes_传递给confusion_matrix()，但是没有用。

y_检验和y_pred如下：y_test =

array([[0, 0, 0, 1, 0, 0],

[0, 0, 0, 0, 1, 0],

[0, 1, 0, 0, 0, 0],

...,

[0, 0, 0, 0, 0, 1],

[0, 0, 0, 1, 0, 0],

[0, 0, 0, 0, 1, 0]])

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, 1],

[0, 0, 0, 0, 0, 1],

[0, 0, 0, 0, 0, 0]])