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 |
```
|