Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
base_model: Alibaba-NLP/gte-reranker-modernbert-base
|
| 4 |
+
tags:
|
| 5 |
+
- cross-encoder
|
| 6 |
+
- reranker
|
| 7 |
+
- sentence-transformers
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# Fine-tuned reranker (best checkpoint, step 300)
|
| 11 |
+
|
| 12 |
+
Fine-tuned version of `Alibaba-NLP/gte-reranker-modernbert-base` on hard-negative-mined query/document pairs from scikit-learn's API documentation.
|
| 13 |
+
|
| 14 |
+
**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.
|
| 15 |
+
|
| 16 |
+
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
|