tensorflow 实战时遇到的一些问题记录

最新推荐文章于 2025-05-04 03:34:55 发布

Pacer95

最新推荐文章于 2025-05-04 03:34:55 发布

阅读量240

点赞数

CC 4.0 BY-SA版权

分类专栏： python tensorflow

本文链接：https://blog.youkuaiyun.com/Pacer95/article/details/84024162

python 同时被 2 个专栏收录

5 篇文章

订阅专栏

tensorflow

2 篇文章

订阅专栏

本文探讨了TensorFlow中tf.argmax函数的使用方法，通过示例展示了如何获取矩阵中每行最大值的真实列号，并计算了预测准确率。了解tf.argmax在深度学习中的应用对于模型评估至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. tf.argmax 返回的是最大值所在的索引号，比如B第一行最大值7在第2列，所以返回的是1，若想得到是真实所在列号，通过tf.argmax(B,1)+1即可

import tensorflow as tf
B=[[2,7,1],[1,3,4],[2,0,1],[1,3,0.4]]
C=[[8,7,1],[2,1,6],[2,1,1],[1.4,53,4]]
accuracy = tf.reduce_mean(tf.cast(correct_pred, "float"))

#print(tf.argmax(x, 1))

with tf.Session() as sess:
    print(sess.run(tf.argmax(B, 1)+1))
    print(sess.run(tf.argmax(C, 1)))
    print(sess.run(accuracy))

输出：[2 3 1 2]
[0 2 0 1]
0.75