sentence-transformers How to use DariaaaS/e5-args-1 with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("DariaaaS/e5-args-1")
sentences = [
"How many athletes from region 151 have won a medal?",
"athletes refer to person_id; region 151 refers to region_id = 151; won a medal refers to medal_id <> 4;",
"Rio de Janeiro refers to city_name = 'Rio de Janeiro';",
"the highest number of participants refers to MAX(COUNT(person_id)); the lowest number of participants refers to MIN(COUNT(person_id)); Which summer Olympic refers to games_name where season = 'Summer';"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
