Update README.md
Browse files
README.md
CHANGED
|
@@ -33,14 +33,14 @@ from promcse import PromCSE
|
|
| 33 |
model = PromCSE("hellonlp/promcse-roberta-base-zh", "cls", 10)
|
| 34 |
```
|
| 35 |
|
| 36 |
-
Then you can use our model for
|
| 37 |
```python
|
| 38 |
embeddings = model.encode("武汉是一个美丽的城市。")
|
| 39 |
print(embeddings.shape)
|
| 40 |
#torch.Size([768])
|
| 41 |
```
|
| 42 |
|
| 43 |
-
|
| 44 |
```python
|
| 45 |
sentences_a = ['你好吗']
|
| 46 |
sentences_b = ['你怎么样','我吃了一个苹果','你过的好吗','你还好吗','你',
|
|
|
|
| 33 |
model = PromCSE("hellonlp/promcse-roberta-base-zh", "cls", 10)
|
| 34 |
```
|
| 35 |
|
| 36 |
+
Then you can use our model for encoding sentences into embeddings
|
| 37 |
```python
|
| 38 |
embeddings = model.encode("武汉是一个美丽的城市。")
|
| 39 |
print(embeddings.shape)
|
| 40 |
#torch.Size([768])
|
| 41 |
```
|
| 42 |
|
| 43 |
+
Compute the cosine similarities between two groups of sentences
|
| 44 |
```python
|
| 45 |
sentences_a = ['你好吗']
|
| 46 |
sentences_b = ['你怎么样','我吃了一个苹果','你过的好吗','你还好吗','你',
|