From ba5c86c0bc8da450dee361140541f35b5349a492 Mon Sep 17 00:00:00 2001 From: Craig Macdonald Date: Wed, 28 Aug 2024 13:02:44 +0100 Subject: [PATCH] Update for PyT 0.11 API --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index e472399..74cae2b 100644 --- a/README.md +++ b/README.md @@ -22,7 +22,7 @@ dense_e2e = pytcolbert.end_to_end() A ColBERT re-ranker of BM25 can be formulated as follows (you will need to have an index with [text saved](https://pyterrier.readthedocs.io/en/latest/text.html) - the Terrier data repostiory conviniently [provides such an index](http://data.terrier.org/msmarco_passage.dataset.html#terrier_stemmed_text)): ```python -bm25 = pt.BatchRetrieve.from_dataset('msmarco_passage', 'terrier_stemmed_text', wmodel='BM25', metadata=['docno', 'text']) +bm25 = pt.terrier.Retriever.from_dataset('msmarco_passage', 'terrier_stemmed_text', wmodel='BM25', metadata=['docno', 'text']) sparse_colbert = bm25 >> pytcolbert.text_scorer() ```