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