Sahajtomar commited on
Commit
31cb2df
1 Parent(s): 63efba5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +53 -1
README.md CHANGED
@@ -1 +1,53 @@
1
- "hello"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: German
3
+ tags:
4
+ - sts
5
+ - pytorch
6
+ - semantic
7
+ - sentence transformer
8
+ - de
9
+ datasets:
10
+ - sts
11
+ pipeline_tag: Semantic Similarity
12
+ ---
13
+ # German STS
14
+
15
+ ## STS dev (german)
16
+ 87.9%
17
+ ## STS test (german)
18
+ 84.3%
19
+
20
+ #### STS pipeline
21
+ ```python
22
+ !pip install -U sentence-transformers
23
+ from sentence_transformers import SentenceTransformer
24
+ model = SentenceTransformer('..model_path..')
25
+ sentences1 = ['Die Katze sitzt draußen',
26
+ "Ein Mann spielt Gitarre",
27
+ 'Der neue Film ist großartig']
28
+
29
+ sentences2 = ['Der Hund spielt im Garten',
30
+ "Eine Frau sieht fern",
31
+ 'Der neue Film ist so toll']
32
+
33
+ #Compute embedding for both lists
34
+ embeddings1 = model.encode(sentences1, convert_to_tensor=True)
35
+ embeddings2 = model.encode(sentences2, convert_to_tensor=True)
36
+
37
+ #Compute cosine-similarits
38
+ cosine_scores = util.pytorch_cos_sim(embeddings1, embeddings2)
39
+
40
+ #Output the pairs with their score
41
+ for i in range(len(sentences1)):
42
+ for j in range(len(sentences2)):
43
+ print("{} \t\t {} \t\t Score: {:.4f}".format(sentences1[i], sentences2[j], cosine_scores[i][j]))
44
+ """
45
+ Die Katze sitzt draußen Der Hund spielt im Garten Score: 0.1259
46
+ Die Katze sitzt draußen Eine Frau sieht fern Score: 0.0567
47
+ Die Katze sitzt draußen Der neue Film ist so toll Score: 0.0557
48
+ Ein Mann spielt Gitarre Der Hund spielt im Garten Score: 0.1031
49
+ Ein Mann spielt Gitarre Eine Frau sieht fern Score: 0.0098
50
+ Ein Mann spielt Gitarre Der neue Film ist so toll Score: 0.0828
51
+ Der neue Film ist großartig Der Hund spielt im Garten Score: 0.1008
52
+ Der neue Film ist großartig Eine Frau sieht fern Score: 0.0674
53
+ """