Views
No views yet
1
2!pip install -U sentence-transformers
3from sentence_transformers import SentenceTransformer
4model = SentenceTransformer('..model_path..')
5sentences1 = ['Die Katze sitzt draußen',
6 "Ein Mann spielt Gitarre",
7 'Der neue Film ist großartig']
8
9sentences2 = ['Der Hund spielt im Garten',
10 "Eine Frau sieht fern",
11 'Der neue Film ist so toll']
12embeddings1 = model.encode(sentences1, convert_to_tensor=True)
13embeddings2 = model.encode(sentences2, convert_to_tensor=True)
14cosine_scores = util.pytorch_cos_sim(embeddings1, embeddings2)
15
16for i in range(len(sentences1)):
17 for j in range(len(sentences2)):
18 print(cosine_scores[i][j]))
19"""
20Die Katze sitzt draußen Der Hund spielt im Garten Score: 0.1259
21Die Katze sitzt draußen Eine Frau sieht fern Score: 0.0567
22Die Katze sitzt draußen Der neue Film ist so toll Score: 0.0557
23Ein Mann spielt Gitarre Der Hund spielt im Garten Score: 0.1031
24Ein Mann spielt Gitarre Eine Frau sieht fern Score: 0.0098
25Ein Mann spielt Gitarre Der neue Film ist so toll Score: 0.0828
26Der neue Film ist großartig Der Hund spielt im Garten Score: 0.1008
27Der neue Film ist großartig Eine Frau sieht fern Score: 0.0674
28"""