张量流：如何创建混淆矩阵

Question

我是tensorflow的新手，我使用了以下教程：

https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/。在

我已经在包含3个标签的新数据集上训练了相同的模型。我正在尝试创建混乱矩阵。在

在tf.混淆矩阵功能非常混乱。在

有人能帮助使用相同的代码示例吗。在

Answer 1

你有3个标签（比如0，1，2）。假设您有一个大小为10的测试集，并且得到以下张量： 真理：[0,0,0,1,1,2,2,2] 预测：[2,0,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)

注意以下几点： 结果是一个3x3矩阵。第一行表示，标签0的2倍预测正确，一次被误认为标签1，一次被误认为标签2。在