Hugging Face's logo

KshitijKrishna
/
pivot-reranker

Text Classification
Safetensors
English
Chinese
xlm-roberta
mteb
Eval Results (legacy)
Model card Files
xet
 Community
pivot-reranker
File size: 2,156 Bytes
3fe8ac7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
---
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)