tf.nn.top_k(input, k, name=None)和tf.nn.in_top_k(predictions, targets, k, name=None)
2024-10-01 15:37:27
tf.nn.top_k(input, k, name=None)
这个函数的作用是返回 input 中每行最大的 k 个数,并且返回它们所在位置的索引。
input: 一个张量,数据类型必须是以下之一:float32、float64、int32、int64、uint8、int16、int8。数据维度是 batch_size 乘上 x 个类别。
k: 一个整型,必须 >= 1。在每行中,查找最大的 k 个值。
name: 为这个操作取个名字。
输出:一个元组 Tensor ,数据元素是 (values, indices),具体如下:
values: 一个张量,数据类型和 input 相同。数据维度是 batch_size 乘上 k 个最大值。
indices: 一个张量,数据类型是 int32 。每个最大值在 input 中的索引位置。
tf.nn.in_top_k(predictions, targets, k, name=None)
就是对比predictions和targets是否一样,一样的返回true,不一样的返回false,接下来用tf.cast(correct,tf.floatxx) 可以计算准确率
predictions:预测的结果,预测矩阵大小为样本数×标注的label类的个数的二维矩阵。
targets:实际的标签,大小为样本数。
k:每个样本的预测结果的前k个最大的数里面是否包含targets预测中的标签,一般都是取1,即取预测最大概率的索引与标签对比。
name:名字。
最新文章
- Skype无法收发组消息
- Linux下搭建个人网站
- jsTree 的简单用法--异步加载和刷新数据
- 剑指offer一:二维数组中的查找
- HTML的标签-W3School读后总结
- 为什么Facebook要将视频从Flash全面迁移到HTML5?
- 九度OJ 1512 用两个栈实现队列 【数据结构】
- java排序算法-选择排序
- Viewing the Raw SQL Statement(xcode で)
- 《JavaScript+DOM编程艺术》的摘要(四)appendChild与insertBefore的区别
- nginx 区分pc和mobile 到不同的404页面
- AngularJS框架研究(一)
- angular $parse $eval parse VS eval
- 成为一名Java架构师的必修课
- Mawawa CSS 学习之旅 Display
- GIS中空间数据和非空间数据
- 显示日期的指令: date
- STL_string.vector中find到的iterator的序号
- mongodb导入json文件
- python3两个字典的合并