davanstrien/code-prompt-similarity-model

Instalar la librería de transformadores de oraciones:

pip install -U sentence-transformers

Luego puedes cargar este modelo y ejecutar inferencias:

from sentence_transformers import SentenceTransformer
# Descargar desde el Hub
model = SentenceTransformer("davanstrien/code-prompt-similarity-model")

# Ejecutar inferencias
sentences = [
    'Escribe una función de Python que toma un número de nota MIDI y devuelve el número de tecla del piano correspondiente.',
    'Crea una función de Python que traduzca números de nota MIDI a números de teclas del piano, facilitando la generación de música.',
    'Escribe una función de Python que acepte un diccionario y devuelva un conjunto de valores distintos. Si una clave mapea a una lista vacía, devuelve un conjunto vacío.',
]
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]