BGE base Finanzas Matryoshka

Instale la biblioteca Sentence Transformers:

pip install -U sentence-transformers

Luego puede cargar este modelo y ejecutar inferencias:

from sentence_transformers import SentenceTransformer

# Descargar del Hub de 🤗
model = SentenceTransformer("Sailesh9999/bge-base-financial-matryoshka_3")

# Ejecutar inferencias
sentences = [
    'Chipotle retains an independent third-party compensation consultant each year to conduct a pay equity analysis of its U.S. and Canadian workforce, including factors of pay such as grade level, tenure in role, and external market conditions like geographic location, to ensure consistency and equitable treatment among employees.',
    'How does Chipotle ensure pay equity among its employees?',
    'How can one locate information on legal proceedings within the Consolidated Financial Statements?',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

# Obtener las puntuaciones de similitud para los embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]