在tensorflow.metrics中, precision_at_k 和 precision_at_top_k 之间的区别?

syl*_*ong 5 python information-retrieval machine-learning tensorflow

在tensorflow python API中,tf.metrics具有一些用于信息检索的指标。

尤其:

  • tf.precision_at_ktf.precision_at_top_k
  • tf.recall_at_ktf.recall_at_top_k

_at_k和指标有什么区别_at_top_k

API文档似乎没有提供这方面的信息。

xdu*_*ch0 4

看看他们的实现precision_at_k是一个简单的包装precision_at_top_k。API 文档中实际上提到了这种差异:precision_at_k期望一个 logits 张量,而predictions期望precision_at_top_k预测是前 k 个类的索引。本质上,precision_at_k只需执行tf.nn.top_kpredictions然后调用precision_at_top_k