阅读(5.3k) 书签 (0)

TensorFlow函数教程:tf.keras.backend.in_top_k

2019-03-13 17:35 更新

tf.keras.backend.in_top_k函数

tf.keras.backend.in_top_k(
    predictions,
    targets,
    k
)

定义在:tensorflow/python/keras/backend.py。

判断targets是否在predictions的前k个中。

参数:

  • predictions:一个张量,shape为(batch_size, classes),类型为float32。
  • targets:一个1D张量,长度为batch_size,类型为int32或int64。
  • k:一个int,要考虑的顶级元素的数量。

返回:

一个 1D 张量,长度为batch_size,类型为bool。如果predictions[i, targets[i]]predictions[i]的 top-k值中,则output[i]True