pivot-reranker / README.md
KshitijKrishna's picture
reranker model
3fe8ac7
|
Raw
History Blame Contribute Delete
2.16 kB
---
license: mit
language:
- en
- zh
tags:
- mteb
model-index:
- name: bge-reranker-base
results:
- task:
type: Reranking
dataset:
type: C-MTEB/CMedQAv1-reranking
name: MTEB CMedQAv1
config: default
split: test
revision: None
metrics:
- type: map
value: 81.27206722525007
- type: mrr
value: 84.14238095238095
- task:
type: Reranking
dataset:
type: C-MTEB/CMedQAv2-reranking
name: MTEB CMedQAv2
config: default
split: test
revision: None
metrics:
- type: map
value: 84.10369934291236
- type: mrr
value: 86.79376984126984
- task:
type: Reranking
dataset:
type: C-MTEB/Mmarco-reranking
name: MTEB MMarcoReranking
config: default
split: dev
revision: None
metrics:
- type: map
value: 35.4600511272538
- type: mrr
value: 34.60238095238095
- task:
type: Reranking
dataset:
type: C-MTEB/T2Reranking
name: MTEB T2Reranking
config: default
split: dev
revision: None
metrics:
- type: map
value: 67.27728847727172
- type: mrr
value: 77.1315192743764
pipeline_tag: text-classification
---
# πŸ” BGE Reranker Base (Fine-tuned)
This repository contains a **fine-tuned cross-encoder reranker model** based on `BAAI/bge-reranker-base`.
It is designed to improve retrieval pipelines by re-ranking top-K results from embedding models.
---
## πŸš€ Key Features
- Cross-encoder reranking (query + passage scoring)
- Strong performance on MTEB / C-MTEB benchmarks
- Supports multilingual (English + Chinese)
- Optimized for semantic search pipelines
---
## πŸ“¦ Model Files
This repo contains:
βœ” config.json
βœ” model.safetensors
βœ” tokenizer_config.json
βœ” sentencepiece.bpe.model
βœ” special_tokens_map.json (fixed naming)
βœ” README.md
## βš™οΈ Usage
### πŸ”Ή Using FlagEmbedding (Recommended)
```python
from FlagEmbedding import FlagReranker
reranker = FlagReranker("BAAI/bge-reranker-base", use_fp16=True)
score = reranker.compute_score(["query", "passage"])
print(score)