Downloads · 30 days
28
4% of all-time downloads
yobi/klue-roberta-base-sts
klue-roberta-base-sts is a sentence similarity model from yobi. Use it when you need a score for how close two texts are. It is set up for sentence-transformers.
Downloads · 30 days
28
4% of all-time downloads
All-time downloads
751
Public
Repo size
885 MB
Likes
0
Public
Click a slice to open those files.
.bin443 MB · 100%
From the Hugging Face model README
from sentence_transformers import SentenceTransformer, models
embedding_model = models.Transformer("yobi/klue-roberta-base-sts")
pooling_model = models.Pooling(
embedding_model.get_word_embedding_dimension(),
pooling_mode_mean_tokens=True,
)
model = SentenceTransformer(modules=[embedding_model, pooling_model])
model.encode("안녕하세요.", convert_to_tensor=True)