sentence-transformers How to use uer/sbert-base-chinese-nli with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("uer/sbert-base-chinese-nli")
sentences = [
"那个人很开心",
"那个人非常开心",
"那只猫很开心",
"那个人在吃东西"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]
