How to use davanstrien/code-prompt-similarity-model with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("davanstrien/code-prompt-similarity-model") sentences = [ "Write a Python function that counts the number of even numbers in a given list of integers or floats", "Write a Python function that returns the number of even numbers in a list.", "Create a Python function that adds up all the numbers in a given list. The function should support lists containing only positive integers.", "Write a Python function that converts a JSON string into a Python dictionary using the json module and returns it." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
