Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Vector PRF #25

Merged
merged 12 commits into from
Nov 26, 2024
Prev Previous commit
Next Next commit
bibtex block in documentation
  • Loading branch information
seanmacavaney committed Nov 25, 2024

Unverified

This user has not yet uploaded their public signing key.
commit da7d1f2827e329d2b0020e44ca5c7e97ae8ded07
46 changes: 44 additions & 2 deletions pyterrier_dr/prf.py
Original file line number Diff line number Diff line change
@@ -21,7 +21,28 @@ class VectorPrf(pt.Transformer):

prf_pipe = model >> index >> index.vec_loader() >> pyterier_dr.vector_prf() >> index

Reference: Hang Li, Ahmed Mourad, Shengyao Zhuang, Bevan Koopman, Guido Zuccon. [Pseudo Relevance Feedback with Deep Language Models and Dense Retrievers: Successes and Pitfalls](https://arxiv.org/pdf/2108.11044.pdf)
.. code-block:: bibtex
:caption: Citation

@article{DBLP:journals/tois/0009MZKZ23,
author = {Hang Li and
Ahmed Mourad and
Shengyao Zhuang and
Bevan Koopman and
Guido Zuccon},
title = {Pseudo Relevance Feedback with Deep Language Models and Dense Retrievers:
Successes and Pitfalls},
journal = {{ACM} Trans. Inf. Syst.},
volume = {41},
number = {3},
pages = {62:1--62:40},
year = {2023},
url = {https://doi.org/10.1145/3570724},
doi = {10.1145/3570724},
timestamp = {Fri, 21 Jul 2023 22:26:51 +0200},
biburl = {https://dblp.org/rec/journals/tois/0009MZKZ23.bib},
bibsource = {dblp computer science bibliography, https://dblp.org}
}
"""
def __init__(self,
*,
@@ -65,7 +86,28 @@ class AveragePrf(pt.Transformer):

prf_pipe = model >> index >> index.vec_loader() >> pyterier_dr.average_prf() >> index

Reference: Hang Li, Ahmed Mourad, Shengyao Zhuang, Bevan Koopman, Guido Zuccon. [Pseudo Relevance Feedback with Deep Language Models and Dense Retrievers: Successes and Pitfalls](https://arxiv.org/pdf/2108.11044.pdf)
.. code-block:: bibtex
:caption: Citation

@article{DBLP:journals/tois/0009MZKZ23,
author = {Hang Li and
Ahmed Mourad and
Shengyao Zhuang and
Bevan Koopman and
Guido Zuccon},
title = {Pseudo Relevance Feedback with Deep Language Models and Dense Retrievers:
Successes and Pitfalls},
journal = {{ACM} Trans. Inf. Syst.},
volume = {41},
number = {3},
pages = {62:1--62:40},
year = {2023},
url = {https://doi.org/10.1145/3570724},
doi = {10.1145/3570724},
timestamp = {Fri, 21 Jul 2023 22:26:51 +0200},
biburl = {https://dblp.org/rec/journals/tois/0009MZKZ23.bib},
bibsource = {dblp computer science bibliography, https://dblp.org}
}
"""
def __init__(self,
*,
Loading