擅长:python、mysql、java
<p>不,您对<a href="http://scikit-learn.org/stable/modules/generated/sklearn.metrics.confusion_matrix.html" rel="noreferrer">^{<cd1>}</a>的输入必须是预测列表,而不是OHEs(一个热编码)。在你的<code>y_test</code>和<code>y_pred</code>上调用<code>argmax</code>,你应该得到你期望的结果。</p>
<pre><code>confusion_matrix(
y_test.values.argmax(axis=1), predictions.argmax(axis=1))
array([[1, 0],
[0, 2]])
</code></pre>