Update README.md
Browse files
README.md
CHANGED
@@ -33,7 +33,7 @@ print(embeddings)
|
|
33 |
```
|
34 |
|
35 |
## Calculate Sentence similarities
|
36 |
-
You can
|
37 |
```python
|
38 |
from sklearn.metrics.pairwise import cosine_similarity
|
39 |
sentences_a = [['Represent the Science sentence; Input: ','Parton energy loss in QCD matter',0],
|
@@ -43,4 +43,5 @@ sentences_b = [['Represent the Science sentence; Input: ','The Chiral Phase Tran
|
|
43 |
embeddings_a = model.encode(sentences_a)
|
44 |
embeddings_b = model.encode(sentences_b)
|
45 |
similarities = cosine_similarity(embeddings_a,embeddings_b)
|
|
|
46 |
```
|
|
|
33 |
```
|
34 |
|
35 |
## Calculate Sentence similarities
|
36 |
+
You can further use the model to compute similarities between two groups of sentences, with **customized embeddings**.
|
37 |
```python
|
38 |
from sklearn.metrics.pairwise import cosine_similarity
|
39 |
sentences_a = [['Represent the Science sentence; Input: ','Parton energy loss in QCD matter',0],
|
|
|
43 |
embeddings_a = model.encode(sentences_a)
|
44 |
embeddings_b = model.encode(sentences_b)
|
45 |
similarities = cosine_similarity(embeddings_a,embeddings_b)
|
46 |
+
print(similarities)
|
47 |
```
|