Sentence Similarity
sentence-transformers
Safetensors
Korean
xlm-roberta
embedding
classification
mteb
korean
text-embeddings-inference
Instructions to use NGA-KR/ko-embed-cls with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use NGA-KR/ko-embed-cls with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("NGA-KR/ko-embed-cls") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
NGA-KR/ko-embed-cls
Korean text embedding model (568M) fine-tuned from BAAI/bge-m3 for topic classification.
Training data
- KLUE-TC (ynat) train split only (45,678 news titles, 7 topics), label-aware contrastive pairs (same-topic titles as positives, other-topic titles as hard negatives).
- No evaluation splits were used. Declared as
KLUE-TCin the MTEB model metadata; base-model training data is inherited from bge-m3.
Evaluation (MTEB(kor, v2))
| Task | Score |
|---|---|
| KLUE-TC (Classification, accuracy) | 84.39 |
Training
CachedMultipleNegativesRankingLoss · batch 512 · lr 2e-5 · 1 epoch · max_len 128 · CLS pooling · bf16
Usage
from sentence_transformers import SentenceTransformer
m = SentenceTransformer("NGA-KR/ko-embed-cls")
e = m.encode(["삼성전자, 2분기 영업이익 10조 돌파", "손흥민 시즌 20호골"])
License
MIT (same as BAAI/bge-m3).
- Downloads last month
- -
Model tree for NGA-KR/ko-embed-cls
Base model
BAAI/bge-m3