pyNeuIR is a toolkit for fast prototyping and reproducing Neu-IR ranking models.
This toolkit is organised in components as shown in this schematic:
- Duet (Local and Distributed version)
- DRMM
All components:
- Python 3.5+
Datasets, Models, Trainers, and Rerankers:
- Pytorch 2.0+
- Gensim
Preprocessors:
- Pyndri