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

Question

我无法从评分函数中获得每一行的文件输出的分数。你知道吗

我正在使用一个评分函数在推荐引擎中对我的文档进行排名，并在Jupyter笔记本中使用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，u，mean和其他度量。现在，我需要一个输出文件，它对我的测试数据的每一行都有分数，就像我们在任何其他ml分类问题中得到预测输出一样。请帮帮我！你知道吗

Answer 1

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

在笔记本的最后，在“生成预测”下