tf.nn.in_top_k()的用法

tf.nn.in_top_k()的用法

      tf.nn.in_top_k()中第一个参数是train data,第二个是labels,k一般取1,函数判断(0.6,0.5,111)中最大的111位于第2位,而labels中是1,显然不对应,返回False,在(0.1,0.6,0.4)中最大值是0.6,位于第1位,和labels中的1对应,则返回True。

      实际上labels大多采用one-hot encoding,这样就刚好对应起来了。