Views
No views yet
| Metric | Base (bge-m3) | Fine-tuned | Delta |
|---|---|---|---|
| Recall@1 | 29.5% | 85.0% | ↑ 55.5% |
| Recall@5 | 63.0% | 96.5% | ↑ 33.5% |
| Recall@10 | 76.0% | 99.0% | ↑ 23.0% |
| MRR | 45.0% | 90.4% | ↑ 45.4% |
| NDCG@10 | 51.8% | 92.5% | ↑ 40.6% |
| Metric | Base (bge-m3) | Fine-tuned | Delta |
|---|---|---|---|
| Accuracy | 37.5% | 94.0% | ↑ 56.5% |
| MRR | 57.3% | 96.6% | ↑ 39.4% |
1from FlagEmbedding import BGEM3FlagModel
2
3model = BGEM3FlagModel("Sophia-AI/bge-m3-bank-it", device="cuda", use_fp16=True)
4
5# Embeddings
6output = model.encode(["Your query here"], return_dense=True, return_sparse=True)
7
8# Reranking
9scores = model.compute_score(
10 [["query", "document"]],
11 weights_for_different_modes=[0.30, 0.65, 0.05],
12)1pip install bge-auto-tune
2
3bge-auto-tune generate --collection your_collection --min-pairs 2000
4bge-auto-tune finetune --dataset bge_m3_training.jsonl --epochs 4
5bge-auto-tune test --model ./bge-m3-finetuned
6bge-auto-tune publish --repo your-user/your-model-name