Downloads · 30 days
14
2% of all-time downloads
ehdwns1516/klue-roberta-base_sae
klue-roberta-base_sae is a text classification model from ehdwns1516. Use it when you need a label for a piece of text. It is set up for transformers.
This model trained with Korean dataset. Input sentence what you want to grasp intent. You can use English, but don't expect accuracy.
Downloads · 30 days
14
2% of all-time downloads
All-time downloads
891
Public
Repo size
885 MB
Likes
0
Public
Click a slice to open those files.
.bin443 MB · 100%
From the Hugging Face model README
klue-roberta-base-kornli DEMO: Ainize DEMO
klue-roberta-base-kornli API: Ainize API
Language model: klue/roberta-base
Language: Korean
Training data: kor_sae
Eval data: kor_sae
Code: See Ainize Workspace
from transformers import AutoTokenizer, pipeline
tokenizer = AutoTokenizer.from_pretrained("ehdwns1516/klue-roberta-base-sae")
classifier = pipeline(
"text-classification",
model="ehdwns1516/klue-roberta-base-kornli",
return_all_scores=True,
)
context = "sentence what you want to grasp intent"
result = dict()
result[0] = classifier(context)[0]