Spaces:
Runtime error
Runtime error
### Putting it all together | |
You can use Doc2Query or Doc2Query-- in an indexing pipeline to build an index of the expanded documents: | |
<div class="pipeline"> | |
<div class="df" title="Document Frame">D</div> | |
<div class="transformer attn" title="Doc2Query or Doc2Query−− Transformer">Doc2Query[−−]</div> | |
<div class="df" title="Document Frame">D</div> | |
<div class="transformer" title="Indexer">Indexer</div> | |
<div class="artefact" title="Doc2Query Index">IDX</div> | |
</div> | |
```python | |
import pyterrier as pt | |
pt.init() | |
import pyterrier_doc2query | |
doc2query = pyterrier_doc2query.Doc2Query(append=True) | |
dataset = pt.get_dataset('irds:msmarco-passage') | |
indexer = pt.IterDictIndexer('./msmarco_psg') | |
indxer_pipe = doc2query >> indexer | |
indxer_pipe.index(dataset.get_corpus_iter()) | |
``` | |
Once you built an index, you can retrieve from it using any retrieval function (often BM25): | |
<div class="pipeline"> | |
<div class="df" title="Query Frame">Q</div> | |
<div class="transformer" title="BM25 Transformer">BM25 Retriever <div class="artefact" title="Doc2Query Index">IDX</div></div> | |
<div class="df" title="Result Frame">R</div> | |
</div> | |
```python | |
bm25 = pt.BatchRetrieve('./msmarco_psg', wmodel="BM25") | |
``` | |
### References & Credits | |
- Rodrigo Nogueira and Jimmy Lin. [From doc2query to docTTTTTquery](https://cs.uwaterloo.ca/~jimmylin/publications/Nogueira_Lin_2019_docTTTTTquery-v2.pdf). | |
- Mitko Gospodinov, Sean MacAvaney, and Craig Macdonald. Doc2Query--: When Less is More. ECIR 2023. | |
- Craig Macdonald, Nicola Tonellotto, Sean MacAvaney, Iadh Ounis. [PyTerrier: Declarative Experimentation in Python from BM25 to Dense Retrieval](https://dl.acm.org/doi/abs/10.1145/3459637.3482013). CIKM 2021. | |