support sentence_transformers b638976
saintway @ Andrews-MacBook-Pro commited on
How to use shibing624/text2vec-base-chinese with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("shibing624/text2vec-base-chinese")
sentences = [
"那是 個快樂的人",
"那是 條快樂的狗",
"那是 個非常幸福的人",
"今天是晴天"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]