How to use tomaarsen/distilbert-base-uncased-wikipedia-sections-triplet with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("tomaarsen/distilbert-base-uncased-wikipedia-sections-triplet") sentences = [ "All charts rank the top 100.", "There are two primary charts: Gaon Album Chart and Gaon Digital Chart.", "Regional Preferente de Cataluña (3): 1999-00, 2002-03, 2008-09.", "Kyūsaku was born in Fukuoka city, Fukuoka prefecture as Sugiyama Naoki." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
