Update README.md
Browse files
README.md
CHANGED
@@ -18,8 +18,12 @@ language:
|
|
18 |
Pre-trained models can be used like this:
|
19 |
```
|
20 |
from sentence_transformers import CrossEncoder
|
21 |
-
model = CrossEncoder('
|
22 |
-
scores = model.predict([('
|
|
|
|
|
|
|
|
|
23 |
```
|
24 |
|
25 |
The model will predict scores for the pairs `('Sentence 1', 'Sentence 2')` and `('Sentence 3', 'Sentence 4')`.
|
|
|
18 |
Pre-trained models can be used like this:
|
19 |
```
|
20 |
from sentence_transformers import CrossEncoder
|
21 |
+
model = CrossEncoder('bongsoo/albert-small-kor-cross-encoder-v1')
|
22 |
+
scores = model.predict([('대한민국', '제주도'), ('대한민국', '서울')])
|
23 |
+
print(scores)
|
24 |
+
```
|
25 |
+
```
|
26 |
+
[0.3342184 0.23371117]
|
27 |
```
|
28 |
|
29 |
The model will predict scores for the pairs `('Sentence 1', 'Sentence 2')` and `('Sentence 3', 'Sentence 4')`.
|