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]
