Sentence Similarity
sentence-transformers
PyTorch
Transformers
bert
text-classification
cross-encoder
text-embeddings-inference
Instructions to use jamescalam/bert-stsb-cross-encoder with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use jamescalam/bert-stsb-cross-encoder with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("jamescalam/bert-stsb-cross-encoder") query = "Which planet is known as the Red Planet?" passages = [ "Venus is often called Earth's twin because of its similar size and proximity.", "Mars, known for its reddish appearance, is often referred to as the Red Planet.", "Jupiter, the largest planet in our solar system, has a prominent red spot.", "Saturn, famous for its rings, is sometimes mistaken for the Red Planet." ] scores = model.predict([(query, passage) for passage in passages]) print(scores) - Transformers
How to use jamescalam/bert-stsb-cross-encoder with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("jamescalam/bert-stsb-cross-encoder") model = AutoModelForSequenceClassification.from_pretrained("jamescalam/bert-stsb-cross-encoder") - Notebooks
- Google Colab
- Kaggle
Quick Links
Augmented SBERT STSb
This is a sentence-transformers cross encoder model.
It is used as a demo model within the NLP for Semantic Search course, for the chapter on In-domain Data Augmentation with BERT.
- Downloads last month
- 255
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("jamescalam/bert-stsb-cross-encoder") model = AutoModelForSequenceClassification.from_pretrained("jamescalam/bert-stsb-cross-encoder")