マルチラベル・インジケータが混同行列

Question

multilabel-indicator is not supportedのためにサポートされていませんが実行しようとしている私が取得エラーメッセージ、次のとおりです。

confusion_matrix(y_test, predictions)

y_testは形状であるDataFrameです：

Horse | Dog | Cat 
1  0  0 
0  1  0 
0  1  0 
...  ... ... 

predictionsはnumpy array：

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

私は少しエラーメッセージを検索しましたが、実際に私が適用できるものが見つかりませんでした。何かヒント？

Answer 1

いいえ、confusion_matrixへの入力は、OHE（1つのホットエンコーディング）ではなく、予測のリストでなければなりません。 y_testとy_predにargmaxと電話すると、あなたが期待するものを得ることができます。

confusion_matrix(y_test.values.argmax(axis=1), predictions.argmax(axis=1)) 

array([[1, 0], 
     [0, 2]]) 

Answer 2

混同行列は、（ワンホットエンコーディングではなく）ラベルのベクトルを取ります。実行する必要があります

confusion_matrix(y_test.values.argmax(axis=1), predictions.argmax(axis=1))