|
--- |
|
language: |
|
- ko |
|
tags: |
|
- sentence_transformers |
|
- cross_encoder |
|
--- |
|
|
|
# Example |
|
```python |
|
from sentence_transformers import CrossEncoder |
|
|
|
model = CrossEncoder('ddobokki/electra-small-sts-cross-encoder') |
|
model.predict(["κ·Έλ
λ ν볡ν΄μ μμλ€.", "κ·Έλ
λ μ겨μ λλ¬Όμ΄ λ¬λ€."]) |
|
-> 0.8206561 |
|
``` |
|
# Dataset |
|
- KorSTS |
|
- Train |
|
- Test |
|
- KLUE STS |
|
- Train |
|
- Test |
|
|
|
# Performance |
|
| Dataset | Pearson corr.|Spearman corr.| |
|
|--|--|--| |
|
| KorSTS(test) + KLUE STS(test) | 0.8528 | 0.8504 | |
|
|
|
# TODO |
|
Using KLUE 1.1 train, dev data |
|
|
|
|
|
|