SJ-Donald/kor-hate-sentence-large

SJ-Donald/kor-hate-sentence-large is pretrained model using follow:

Models

• beomi/kcbert-large

Datasets

• SJ-Donald/kor-hate-sentence

How to use

from transformers import TextClassificationPipeline, BertForSequenceClassification, AutoTokenizer+

model_name = 'SJ-Donald/kor-hate-sentence-large'
model = BertForSequenceClassification.from_pretrained(model_name)

tokenizer = AutoTokenizer.from_pretrained(model_name)

pipe = TextClassificationPipeline(
    model = model,
    tokenizer = tokenizer,
    device = 0,   # cpu: -1, gpu: gpu number
    return_all_scores = True,
    function_to_apply = 'sigmoid'
)

for result in pipe("이딴 게임할 거면 방송 그만해라 어휴")[0]:
    print(result)

{'label': 'hate', 'score': 0.016597675159573555}
{'label': 'clean', 'score': 0.9842987060546875}