1

我无法从评分函数中获取每行的分数作为输出。

我正在使用评分函数在 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_mean 和其他指标。现在我需要输出文件对我的测试数据的每一行都有分数，就像我们在任何其他 ml 分类问题中得到预测输出一样。请帮忙！

4

1 回答 1

1

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

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

于 2019-10-23T14:51:39.837 回答