Views
No views yet
pip install -U sentence-transformers1from sentence_transformers import SentenceTransformer
2sentences_1 = ["样例数据-1", "样例数据-2"]
3sentences_2 = ["样例数据-3", "样例数据-4"]
4model = SentenceTransformer('lier007/xiaobu-embedding-v2')
5embeddings_1 = model.encode(sentences_1, normalize_embeddings=True)
6embeddings_2 = model.encode(sentences_2, normalize_embeddings=True)
7similarity = embeddings_1 @ embeddings_2.T
8print(similarity)