sentence-transformers How to use sergeyzh/rubert-base-retriever with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("sergeyzh/rubert-base-retriever")
sentences = [
"Это счастливый человек",
"Это счастливая собака",
"Это очень счастливый человек",
"Сегодня солнечный день"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]
