中文

Pytrec_eval:极速的 trec_eval Python 接口

信息检索 2018-06-06 v2

摘要

我们介绍 pytrec_eval,一个面向 trec_eval 信息检索评测工具包的 Python 接口。pytrec_eval 以原生扩展的形式在 Python 中暴露了 trec_eval 的参考实现。我们表明,pytrec_eval 比从 Python 内部作为子进程调用 trec_eval 快约一个数量级。与 NDCG 的原生 Python 实现相比,对于实际规模的排序,pytrec_eval 快一倍。最后,我们展示了其在一个应用中的有效性:将 pytrec_eval 与 Pyndri 和 OpenAI Gym 结合,利用 Q-learning 学习查询扩展。

关键词

引用

@article{arxiv.1805.01597,
  title  = {Pytrec_eval: An Extremely Fast Python Interface to trec_eval},
  author = {Christophe Van Gysel and Maarten de Rijke},
  journal= {arXiv preprint arXiv:1805.01597},
  year   = {2018}
}

备注

SIGIR '18. The 41st International ACM SIGIR Conference on Research & Development in Information Retrieval