Feature Extraction
Transformers
PyTorch
ONNX
Safetensors
English
Chinese
xlm-roberta
text-classification
mteb
Eval Results (legacy)
text-embeddings-inference
Instructions to use BAAI/bge-reranker-large with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use BAAI/bge-reranker-large with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="BAAI/bge-reranker-large")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("BAAI/bge-reranker-large") model = AutoModelForSequenceClassification.from_pretrained("BAAI/bge-reranker-large") - Inference
- Notebooks
- Google Colab
- Kaggle
score_ranges 0~1?
#14
by duzhihua - opened
- use FlagReranker
from FlagEmbedding import FlagReranker
reranker = FlagReranker('BAAI/bge-reranker-large', use_fp16=True) # Setting use_fp16 to True speeds up computation with a slight performance degradation
score = reranker.compute_score(['query', 'passage'])
print(score)
--- 1.513292670249939
2.but use huggingface -> Model card -> Inference API
input:
I like you. I love you
output:
[
[
{
"label": "LABEL_0",
"score": 0.9471544623374939
}
]
]
huggingface uses a sigmoid function to normlize the scores.
You can refer to this discusstion: https://huggingface.co/BAAI/bge-reranker-base/discussions/17