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 = ['你怎么样','我吃了一个苹果','你过的好吗','你还好吗','你',
|