How to use from the
Use from the
sentence-transformers library
from sentence_transformers import CrossEncoder

model = CrossEncoder("ddobokki/electra-small-sts-cross-encoder")

query = "Which planet is known as the Red Planet?"
passages = [
	"Venus is often called Earth's twin because of its similar size and proximity.",
	"Mars, known for its reddish appearance, is often referred to as the Red Planet.",
	"Jupiter, the largest planet in our solar system, has a prominent red spot.",
	"Saturn, famous for its rings, is sometimes mistaken for the Red Planet."
]

scores = model.predict([(query, passage) for passage in passages])
print(scores)

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

Downloads last month
9
Safetensors
Model size
14.1M params
Tensor type
I64
ยท
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support