首页 > 解决方案 > tf.metrics.average_precision_at_k 在 TensorFlow 中是如何工作的?

问题描述

我有以下声明:

labels = tf.constant([1, 2, 5, 7, 3], dtype=tf.int64)
pred = tf.constant([1, 8, 2, 11, 3], dtype=tf.int64)

当我跑

metric = tf.metrics.average_precision_at_k(labels, pred, 5)
sess = tf.Session()
sess.run(tf.local_variables_initializer())
sess.run(metric)

我得到mean_average_precision一个0.55

但是如果我创建以下常量

labels = tf.constant([80354115, 60265163, 10138163, 90299492, 10197671], dtype=tf.int64)
pred = tf.constant([80354115, 30297014, 60265163, 60296843, 10197671], dtype=tf.int64)

我得到一个mean_average_precision0.0这是为什么?

标签: tensorflow

解决方案


推荐阅读