Tensorflow：如何创建混淆矩阵
Tensorflow: How to create confusion matrix
我是tensorflow的新手，我用过这个教程：
https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/。
我在包含 3 个标签的新数据集上训练了相同的模型。我正在尝试创建混淆矩阵。
tf.confusion_matrix 函数很混乱。
有人可以帮忙使用相同的代码示例吗？
您有 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。
我是tensorflow的新手，我用过这个教程：
https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/。
我在包含 3 个标签的新数据集上训练了相同的模型。我正在尝试创建混淆矩阵。
tf.confusion_matrix 函数很混乱。
有人可以帮忙使用相同的代码示例吗？
您有 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。