BGE base Financial Matryoshka

Primero, instale la biblioteca de Sentence Transformers:

pip install -U sentence-transformers

Luego puede cargar este modelo y ejecutar la inferencia:

from sentence_transformers import SentenceTransformer

# Descargar del Hub 🤗
model = SentenceTransformer("korruz/bge-base-financial-matryoshka")
# Ejecutar inferencia
sentences = [
'We use raw materials that are subject to price volatility caused by weather, supply conditions, political and economic variables and other unpredictable factors. We may use futures, options and swap contracts to manage the volatility related to the above exposures.',
'What financial instruments does the company use to manage commodity price exposure?',
'What types of legal proceedings is the company currently involved in?',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

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