python - Tensorflow:如何创建混淆矩阵

Question

我是 tensorflow 新手，我使用了本教程:

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

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

tf.confusion_matrix 函数非常困惑。

有人可以帮忙使用相同的代码示例吗？

Answer 1

您有 3 个标签(例如 0、1、2)。假设您有一个大小为 10 的测试集，并且获得以下张量:真相:[0,0,0,0,1,1,2,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。