如何从 Tensorflow 排名系统的输出文件中获取分数？

Question

我无法从评分函数中获取文件中每一行的输出分数。

我在 Jupyter notebook 中使用评分函数对推荐引擎中的文档进行排名 Python。我正在使用以下代码来评估我的测试数据 -

def eval_metric_fns():
metric_fns = {}
  metric_fns.update({
      "metric/ndcg@%d" % topn: tfr.metrics.make_ranking_metric_fn(
          tfr.metrics.RankingMetricKey.NDCG, topn=topn)
      for topn in [1, 3, 5, 10]
  })

  return metric_fns


ranker.evaluate(input_fn=lambda: input_fn(_TEST_DATA_PATH), steps=100)

上面的代码给了我 logits_mean 和其他指标。现在我需要输出文件对我的测试数据的每一行都有分数，就像我们在任何其他 ml 分类问题中获得预测输出一样。请帮忙！

Answer 1

不确定您是否仍在此处寻找答案，但您现在可以在此处找到有关如何生成测试预测的示例：https://github.com/tensorflow/ranking/blob/master/tensorflow_ranking/examples/handling_sparse_features.ipynb

它在笔记本的最后，'Generating Predictions'

下面

如何从 Tensorflow 排名系统的输出文件中获取分数？

How to get scores in output file from Tensorflow ranking sysytem?

python

tensorflow

recommendation-engine

ranking-functions