eval_metrics.py 文件源码

def _top_k_generator(k):
  def _top_k(probabilities, targets):
    return metric_ops.streaming_mean(nn.in_top_k(probabilities,
                                                 math_ops.to_int32(targets), k))
  return _top_k