├── .gitignore ├── LICENSE ├── Makefile ├── README.md ├── docs ├── .pages ├── CNAME ├── api │ ├── .pages │ ├── losses │ │ ├── .pages │ │ ├── Flops.md │ │ ├── FlopsScheduler.md │ │ └── Ranking.md │ ├── models │ │ ├── .pages │ │ ├── ColBERT.md │ │ ├── SparseEmbed.md │ │ └── Splade.md │ ├── overview.md │ ├── rank │ │ ├── .pages │ │ └── ColBERT.md │ ├── retrieve │ │ ├── .pages │ │ ├── SparseEmbed.md │ │ ├── Splade.md │ │ └── TfIdf.md │ ├── train │ │ ├── .pages │ │ ├── train-colbert.md │ │ ├── train-sparse-embed.md │ │ └── train-splade.md │ └── utils │ │ ├── .pages │ │ ├── batchify.md │ │ ├── colbert-scores.md │ │ ├── dense-scores.md │ │ ├── evaluate.md │ │ ├── in-batch-sparse-scores.md │ │ ├── iter.md │ │ ├── load-beir.md │ │ ├── pairs-dense-scores.md │ │ └── sparse-scores.md ├── css │ └── version-select.css ├── evaluate │ ├── .pages │ └── evaluate.md ├── fine_tune │ ├── .pages │ ├── colbert.md │ ├── sparse_embed.md │ └── splade.md ├── img │ └── logo.png ├── index.md ├── javascripts │ └── config.js ├── js │ └── version-select.js ├── pre_trained_models │ ├── .pages │ └── pre_trained_models.md ├── retrieve │ ├── .pages │ ├── bm25.md │ ├── colbert.md │ ├── sparse_embed.md │ ├── splade.md │ └── tfidf.md ├── scoring │ ├── .pages │ └── scoring.md ├── scripts │ └── index.py └── stylesheets │ └── extra.css ├── mkdocs.yml ├── neural_cherche ├── __init__.py ├── __version__.py ├── losses │ ├── __init__.py │ ├── flops.py │ └── ranking.py ├── models │ ├── __init__.py │ ├── base.py │ ├── colbert.py │ ├── sparse_embed.py │ └── splade.py ├── rank │ ├── __init__.py │ ├── colbert.py │ └── sparse_embed.py ├── retrieve │ ├── __init__.py │ ├── bm25.py │ ├── colbert.py │ ├── sparse_embed.py │ ├── splade.py │ └── tfidf.py ├── train │ ├── __init__.py │ ├── train_colbert.py │ ├── train_sparse_embed.py │ └── train_splade.py └── utils │ ├── __init__.py │ ├── colbert_scores.py │ ├── dense_scores.py │ ├── evaluate.py │ ├── freeze.py │ ├── iter.py │ ├── sparse_scores.py │ └── warnings.py ├── pytest.ini ├── setup.cfg └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/README.md -------------------------------------------------------------------------------- /docs/.pages: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/.pages -------------------------------------------------------------------------------- /docs/CNAME: -------------------------------------------------------------------------------- 1 | raphaelsty.github.io/neural-cherche/ -------------------------------------------------------------------------------- /docs/api/.pages: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/.pages -------------------------------------------------------------------------------- /docs/api/losses/.pages: -------------------------------------------------------------------------------- 1 | title: losses -------------------------------------------------------------------------------- /docs/api/losses/Flops.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/losses/Flops.md -------------------------------------------------------------------------------- /docs/api/losses/FlopsScheduler.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/losses/FlopsScheduler.md -------------------------------------------------------------------------------- /docs/api/losses/Ranking.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/losses/Ranking.md -------------------------------------------------------------------------------- /docs/api/models/.pages: -------------------------------------------------------------------------------- 1 | title: models -------------------------------------------------------------------------------- /docs/api/models/ColBERT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/models/ColBERT.md -------------------------------------------------------------------------------- /docs/api/models/SparseEmbed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/models/SparseEmbed.md -------------------------------------------------------------------------------- /docs/api/models/Splade.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/models/Splade.md -------------------------------------------------------------------------------- /docs/api/overview.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/overview.md -------------------------------------------------------------------------------- /docs/api/rank/.pages: -------------------------------------------------------------------------------- 1 | title: rank -------------------------------------------------------------------------------- /docs/api/rank/ColBERT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/rank/ColBERT.md -------------------------------------------------------------------------------- /docs/api/retrieve/.pages: -------------------------------------------------------------------------------- 1 | title: retrieve -------------------------------------------------------------------------------- /docs/api/retrieve/SparseEmbed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/retrieve/SparseEmbed.md -------------------------------------------------------------------------------- /docs/api/retrieve/Splade.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/retrieve/Splade.md -------------------------------------------------------------------------------- /docs/api/retrieve/TfIdf.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/retrieve/TfIdf.md -------------------------------------------------------------------------------- /docs/api/train/.pages: -------------------------------------------------------------------------------- 1 | title: train -------------------------------------------------------------------------------- /docs/api/train/train-colbert.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/train/train-colbert.md -------------------------------------------------------------------------------- /docs/api/train/train-sparse-embed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/train/train-sparse-embed.md -------------------------------------------------------------------------------- /docs/api/train/train-splade.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/train/train-splade.md -------------------------------------------------------------------------------- /docs/api/utils/.pages: -------------------------------------------------------------------------------- 1 | title: utils -------------------------------------------------------------------------------- /docs/api/utils/batchify.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/batchify.md -------------------------------------------------------------------------------- /docs/api/utils/colbert-scores.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/colbert-scores.md -------------------------------------------------------------------------------- /docs/api/utils/dense-scores.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/dense-scores.md -------------------------------------------------------------------------------- /docs/api/utils/evaluate.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/evaluate.md -------------------------------------------------------------------------------- /docs/api/utils/in-batch-sparse-scores.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/in-batch-sparse-scores.md -------------------------------------------------------------------------------- /docs/api/utils/iter.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/iter.md -------------------------------------------------------------------------------- /docs/api/utils/load-beir.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/load-beir.md -------------------------------------------------------------------------------- /docs/api/utils/pairs-dense-scores.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/pairs-dense-scores.md -------------------------------------------------------------------------------- /docs/api/utils/sparse-scores.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/api/utils/sparse-scores.md -------------------------------------------------------------------------------- /docs/css/version-select.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/css/version-select.css -------------------------------------------------------------------------------- /docs/evaluate/.pages: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/evaluate/.pages -------------------------------------------------------------------------------- /docs/evaluate/evaluate.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/evaluate/evaluate.md -------------------------------------------------------------------------------- /docs/fine_tune/.pages: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/fine_tune/.pages -------------------------------------------------------------------------------- /docs/fine_tune/colbert.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/fine_tune/colbert.md -------------------------------------------------------------------------------- /docs/fine_tune/sparse_embed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/fine_tune/sparse_embed.md -------------------------------------------------------------------------------- /docs/fine_tune/splade.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/fine_tune/splade.md -------------------------------------------------------------------------------- /docs/img/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/img/logo.png -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/javascripts/config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/javascripts/config.js -------------------------------------------------------------------------------- /docs/js/version-select.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/js/version-select.js -------------------------------------------------------------------------------- /docs/pre_trained_models/.pages: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/pre_trained_models/.pages -------------------------------------------------------------------------------- /docs/pre_trained_models/pre_trained_models.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/pre_trained_models/pre_trained_models.md -------------------------------------------------------------------------------- /docs/retrieve/.pages: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/retrieve/.pages -------------------------------------------------------------------------------- /docs/retrieve/bm25.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/retrieve/bm25.md -------------------------------------------------------------------------------- /docs/retrieve/colbert.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/retrieve/colbert.md -------------------------------------------------------------------------------- /docs/retrieve/sparse_embed.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/retrieve/sparse_embed.md -------------------------------------------------------------------------------- /docs/retrieve/splade.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/retrieve/splade.md -------------------------------------------------------------------------------- /docs/retrieve/tfidf.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/retrieve/tfidf.md -------------------------------------------------------------------------------- /docs/scoring/.pages: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/scoring/.pages -------------------------------------------------------------------------------- /docs/scoring/scoring.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/scoring/scoring.md -------------------------------------------------------------------------------- /docs/scripts/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/scripts/index.py -------------------------------------------------------------------------------- /docs/stylesheets/extra.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/docs/stylesheets/extra.css -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /neural_cherche/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/__init__.py -------------------------------------------------------------------------------- /neural_cherche/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/__version__.py -------------------------------------------------------------------------------- /neural_cherche/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/losses/__init__.py -------------------------------------------------------------------------------- /neural_cherche/losses/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/losses/flops.py -------------------------------------------------------------------------------- /neural_cherche/losses/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/losses/ranking.py -------------------------------------------------------------------------------- /neural_cherche/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/models/__init__.py -------------------------------------------------------------------------------- /neural_cherche/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/models/base.py -------------------------------------------------------------------------------- /neural_cherche/models/colbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/models/colbert.py -------------------------------------------------------------------------------- /neural_cherche/models/sparse_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/models/sparse_embed.py -------------------------------------------------------------------------------- /neural_cherche/models/splade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/models/splade.py -------------------------------------------------------------------------------- /neural_cherche/rank/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/rank/__init__.py -------------------------------------------------------------------------------- /neural_cherche/rank/colbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/rank/colbert.py -------------------------------------------------------------------------------- /neural_cherche/rank/sparse_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/rank/sparse_embed.py -------------------------------------------------------------------------------- /neural_cherche/retrieve/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/retrieve/__init__.py -------------------------------------------------------------------------------- /neural_cherche/retrieve/bm25.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/retrieve/bm25.py -------------------------------------------------------------------------------- /neural_cherche/retrieve/colbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/retrieve/colbert.py -------------------------------------------------------------------------------- /neural_cherche/retrieve/sparse_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/retrieve/sparse_embed.py -------------------------------------------------------------------------------- /neural_cherche/retrieve/splade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/retrieve/splade.py -------------------------------------------------------------------------------- /neural_cherche/retrieve/tfidf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/retrieve/tfidf.py -------------------------------------------------------------------------------- /neural_cherche/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/train/__init__.py -------------------------------------------------------------------------------- /neural_cherche/train/train_colbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/train/train_colbert.py -------------------------------------------------------------------------------- /neural_cherche/train/train_sparse_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/train/train_sparse_embed.py -------------------------------------------------------------------------------- /neural_cherche/train/train_splade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/train/train_splade.py -------------------------------------------------------------------------------- /neural_cherche/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/utils/__init__.py -------------------------------------------------------------------------------- /neural_cherche/utils/colbert_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/utils/colbert_scores.py -------------------------------------------------------------------------------- /neural_cherche/utils/dense_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/utils/dense_scores.py -------------------------------------------------------------------------------- /neural_cherche/utils/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/utils/evaluate.py -------------------------------------------------------------------------------- /neural_cherche/utils/freeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/utils/freeze.py -------------------------------------------------------------------------------- /neural_cherche/utils/iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/utils/iter.py -------------------------------------------------------------------------------- /neural_cherche/utils/sparse_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/utils/sparse_scores.py -------------------------------------------------------------------------------- /neural_cherche/utils/warnings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/neural_cherche/utils/warnings.py -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/pytest.ini -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- 1 | # Inside of setup.cfg 2 | [metadata] 3 | description-file = README.md -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/raphaelsty/neural-cherche/HEAD/setup.py --------------------------------------------------------------------------------