ddobokki's picture
Update README.md
ec59c0d
|
raw
history blame
543 Bytes
---
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