Tensorflow：</p> <p><a href="https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/" rel="nofollow noreferrer">https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/</a>：どのように混同行列

Question

を作成する私はtensorflowに新しいです、私はこのチュートリアルを使用しました。

3つのラベルを含む新しいデータセットで同じモデルを訓練しました。私は混乱行列を作成しようとしています。

tf.confusion_matrix関数は非常に混乱します。

誰かが同じコード例を使って助けてもらえますか？

Answer 1

3つのラベル（たとえば、0,1,2）があります。 真理値：[0,0,0,0,1,1,2,2,2,2] 予測値：[2,0、...] 結果は、3×3の行列である：0,1,1,1,2,1,2,2] その後、次の

>>> import tensorflow as tf 
>>> truth = [0,0,0,0,1,1,2,2,2,2] 
>>> prediction = [2,0,0,1,1,1,2,1,2,2] 
>>> cm = tf.contrib.metrics.confusion_matrix(truth, prediction) 
>>> with tf.Session() as sess: 
...  sess.run(cm) 
... 
array([[2, 1, 1], 
     [0, 2, 0], 
     [0, 1, 3]], dtype=int32) 

、として注意を行うことができます。最初の行では、ラベル1と誤ってラベル2と誤解された場合、ラベル0の2回が正しく予測されたと表示されます。