ddobokki's picture
Update README.md
ec59c0d
|
raw
history blame contribute delete
No virus
543 Bytes
metadata
language:
  - ko
tags:
  - sentence_transformers
  - cross_encoder

Example

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