├── .github └── workflows │ └── test.yml ├── .gitignore ├── LICENSE.txt ├── README.md ├── bm25_pt ├── __init__.py └── bm25.py ├── requirements.txt ├── setup.py └── test ├── basic_test.py ├── real_test.py ├── scrolls_test_corpus.p └── scrolls_test_queries.p /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | build/ 2 | dist/ 3 | 4 | __pycache__/ 5 | *.egg-info/ 6 | -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/README.md -------------------------------------------------------------------------------- /bm25_pt/__init__.py: -------------------------------------------------------------------------------- 1 | from .bm25 import BM25 -------------------------------------------------------------------------------- /bm25_pt/bm25.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/bm25_pt/bm25.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | scipy 2 | torch 3 | transformers 4 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/setup.py -------------------------------------------------------------------------------- /test/basic_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/test/basic_test.py -------------------------------------------------------------------------------- /test/real_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/test/real_test.py -------------------------------------------------------------------------------- /test/scrolls_test_corpus.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/test/scrolls_test_corpus.p -------------------------------------------------------------------------------- /test/scrolls_test_queries.p: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jxmorris12/bm25_pt/HEAD/test/scrolls_test_queries.p --------------------------------------------------------------------------------