sentence-transformers

# Fast, good quality (384 dim)
model = SentenceTransformer('all-MiniLM-L6-v2')
 
# Better quality (768 dim)
model = SentenceTransformer('all-mpnet-base-v2')
 
# Best quality (1024 dim, slower)
model = SentenceTransformer('all-roberta-large-v1')

# 50+ languages
model = SentenceTransformer('paraphrase-multilingual-MiniLM-L12-v2')
 
# 100+ languages
model = SentenceTransformer('paraphrase-multilingual-mpnet-base-v2')

# Legal domain
model = SentenceTransformer('nlpaueb/legal-bert-base-uncased')
 
# Scientific papers
model = SentenceTransformer('allenai/specter')
 
# Code
model = SentenceTransformer('microsoft/codebert-base')

from sentence_transformers import SentenceTransformer, util
 
model = SentenceTransformer('all-MiniLM-L6-v2')
 
# Corpus
corpus = [
    "Python is a programming language",
    "Machine learning uses algorithms",
    "Neural networks are powerful"
]
 
# Encode corpus
corpus_embeddings = model.encode(corpus, convert_to_tensor=True)
 
# Query
query = "What is Python?"
query_embedding = model.encode(query, convert_to_tensor=True)
 
# Find most similar
hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=3)
print(hits)

# Cosine similarity
similarity = util.cos_sim(embedding1, embedding2)
 
# Dot product
similarity = util.dot_score(embedding1, embedding2)
 
# Pairwise cosine similarity
similarities = util.cos_sim(embeddings, embeddings)

# Efficient batch processing
sentences = ["sentence 1", "sentence 2", ...] * 1000
 
embeddings = model.encode(
    sentences,
    batch_size=32,
    show_progress_bar=True,
    convert_to_tensor=False  # or True for PyTorch tensors
)

from sentence_transformers import InputExample, losses
from torch.utils.data import DataLoader
 
# Training data
train_examples = [
    InputExample(texts=['sentence 1', 'sentence 2'], label=0.8),
    InputExample(texts=['sentence 3', 'sentence 4'], label=0.3),
]
 
train_dataloader = DataLoader(train_examples, batch_size=16)
 
# Loss function
train_loss = losses.CosineSimilarityLoss(model)
 
# Train
model.fit(
    train_objectives=[(train_dataloader, train_loss)],
    epochs=10,
    warmup_steps=100
)
 
# Save
model.save('my-finetuned-model')

from langchain_community.embeddings import HuggingFaceEmbeddings
 
embeddings = HuggingFaceEmbeddings(
    model_name="sentence-transformers/all-mpnet-base-v2"
)
 
# Use with vector stores
from langchain_chroma import Chroma
 
vectorstore = Chroma.from_documents(
    documents=docs,
    embedding=embeddings
)

from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 
embed_model = HuggingFaceEmbedding(
    model_name="sentence-transformers/all-mpnet-base-v2"
)
 
from llama_index.core import Settings
Settings.embed_model = embed_model
 
# Use in index
index = VectorStoreIndex.from_documents(documents)