Views
No views yet
1from sentence_transformers import SentenceTransformer
2from sklearn.metrics.pairwise import cosine_similarity
3query = ['그는 그녀를 좋아한다.']
4sentences = ["he love her", "he hate her", '그녀는 그를 싫어한다.','attention is all you need']
5
6emb1 = model.encode(query)
7emb2 = model.encode(sentences)
8print(cosine_similarity(emb1,emb2))
9-> array([[0.62751913, 0.23996451, 0.30788696, 0.08123618]], dtype=float32)