Downloads · 30 days
8
24% of all-time downloads
Sgupta02/code-search-reranker-finetuned
code-search-reranker-finetuned is a machine learning model from Sgupta02. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for sentence-transformers. The card lists the license as apache-2.0.
Fine-tuned version of Alibaba-NLP/gte-reranker-modernbert-base on hard-negative-mined query/document pairs from scikit-learn's API documentation.
Downloads · 30 days
8
24% of all-time downloads
All-time downloads
34
Public
Parameters
150M
598 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors598 MB · 99%
From the Hugging Face model README
Fine-tuned version of Alibaba-NLP/gte-reranker-modernbert-base on hard-negative-mined query/document pairs from scikit-learn's API documentation.
Honest result: this does not beat the pretrained base model. Best checkpoint (step 300 of 1194, learning rate 2e-6) scored 0.534 nDCG@10 on a 26-query eval set, versus 0.556 for the untouched pretrained model. Included here as the actual artifact from a real fine-tuning attempt, not as a recommended model to use.
Full pipeline, evaluation methodology, and the stage-by-stage ablation (dense retrieval to hybrid search to reranking) that this was built on top of: https://github.com/sgupta2346/Code-Search-Finetuned-Reranker