search

terrier-org / pyterrier

A Python framework for performing information retrieval experiments, building on http://terrier.org/
https://pyterrier.readthedocs.io/
Mozilla Public License 2.0
397 stars 63 forks source link

Add RankVicuna and RankZephyr Plugins #441

Closed kaustubhdhole closed 2 months ago

seanmacavaney commented 2 months ago

Awesome, thanks @kaustubhdhole!

A quick tip: If you change LLMReRanker to inherit from pt.Transformer and rename rerank_pyt to transform, you can avoid the lambda. And if you use pt.text.get_text() in the pipeline, it'll add the document's text into the dataframe, so you don't need to load them all into memory.

kaustubhdhole commented 2 months ago

Thanks @seanmacavaney for merging and for the suggestion! I will update the other repo.