Downloads · 30 days
113
6% of all-time downloads
klue-roberta-base-nli-sts-ko-en
ddobokki/klue-roberta-base-nli-sts-ko-en
klue-roberta-base-nli-sts-ko-en is a sentence similarity model from ddobokki. Use it when you need a score for how close two texts are. It is set up for sentence-transformers.
한국어와 영어의 nli, sts데이터를 klue/roberta-base에 학습시킨 모델입니다.
All-time downloads
1.8K
Public
Parameters
111M
885 MB on disk
Likes
2
Public
Repo makeup
Click a slice to open those files.
.bin443 MB · 50%
Parameter types
How the weights are stored.
F32111M · 100%
Model card
From the Hugging Face model README
한국어와 영어의 nli, sts데이터를 klue/roberta-base에 학습시킨 모델입니다.
from sentence_transformers import SentenceTransformer
from sklearn.metrics.pairwise import cosine_similarity
query = ['그는 그녀를 좋아한다.']
sentences = ["he love her", "he hate her", '그녀는 그를 싫어한다.','attention is all you need']
emb1 = model.encode(query)
emb2 = model.encode(sentences)
print(cosine_similarity(emb1,emb2))
-> array([[0.62751913, 0.23996451, 0.30788696, 0.08123618]], dtype=float32)
- Task
- Sentence Similarity
- Library
- sentence-transformers
- Type
- roberta
- Created
- Feb 22, 2023
- Updated
- Nov 8, 2024