🏷️ Text Classification

bge-reranker-base

BAAI/bge-reranker-base

Get AI Model →
2.6M
Downloads
❤️
232
Likes
🏷️
20
Tags
📦
sentence-transformers
Library
Model Details
Full Model IDBAAI/bge-reranker-base
Pipeline / Tasktext-classification
Librarysentence-transformers
Downloads (all-time)2.6M
Likes232
Last Modified6/24/2024
Author / OrgBAAI
PrivateNo — public
⚡ Quick Usage (Python)

Using the 🤗 Transformers library. Install with pip install transformers

from transformers import pipeline

# Load the model
pipe = pipeline("text-classification", model="BAAI/bge-reranker-base")

# Run inference
result = pipe("Your input here")
print(result)
🏷️ Tags
sentence-transformerspytorchonnxsafetensorsxlm-robertamtebtext-embeddings-inferencetext-classificationenzharxiv:2401.03462arxiv:2312.15503arxiv:2311.13534arxiv:2310.07554arxiv:2309.07597license:mitmodel-indexendpoints_compatibledeploy:azureregion:us
More Text Classification Models
See all →
ms-marco-MiniLM-L6-v2

cross-encoder/ms-marco-MiniLM-L6-v2

24.1M❤️ 220
Get AI Model →
bge-reranker-v2-m3

BAAI/bge-reranker-v2-m3

7.5M❤️ 955
Get AI Model →
finbert

ProsusAI/finbert

4.8M❤️ 1.1K
Get AI Model →
🚀 Use This Model

Access model files, inference API, and full documentation on Hugging Face.

Open on Hugging Face →Browse Model Files ↗← Browse All Models
🏷️ Task: Text Classification

This model is designed for the Text Classification task. Explore more models for this use case.

All Text Classification Models →
📊 Popularity
Downloads2.6M
❤️ Community Likes232
🛠️ Requirements
  • Install: pip install sentence-transformers
  • Python 3.8+ recommended for Transformers.
  • GPU (CUDA) speeds up inference significantly.
  • Use model.half() for fp16 on limited VRAM.
👋 Need help with code?