Dense Retrieval for PyTerrier

pyterrier-dr is a PyTerrier extension that provides single-vector [1] dense retrieval functionality through components for dense encoding, indexing, and retrieval.

You can install this extension with pip:

pip install pyterrier-dr

Some functionality requires the installation ot other software packages. For instance, to retrieve using FAISS (e.g., using faiss_hnsw_retriever()), you will need to install the FAISS package:

With pipWith condaWith GPU support

pip install faiss-cpu

Contents

• Overview
• Encoding
• Indexing & Retrieval
• Pseudo-Relevance Feedback
• Diversity
• How-To Guides
• API Reference

---

[1]

If you are interested in multi-vector dense retrieval, check out pyterrier-colbert.