我是 tensorflow 的新手，我使用了本教程：
https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/。
我已经在包含 3 个标签的新数据集上训练了相同的模型。我正在尝试创建混淆矩阵。
tf.confusion_matrix 函数很混乱。
有人可以帮助使用相同的代码示例。
你有 3 个标签（比如 0,1,2）。假设您有一个大小为 10 的测试集，并且您得到以下张量： truth: [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。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句