terrier-org / pyterrier

A Python framework for performing information retrieval experiments, building on http://terrier.org/

Home Page:https://pyterrier.readthedocs.io/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

faster version of add_ranks for single queries

cmacdonald opened this issue · comments

Is your feature request related to a problem? Please describe.
pt.model.add_ranks() can be slow for large transformers, as its sorting by qid and score.

Describe the solution you'd like
An option for add_ranks() that says only one query is being considered.