metadata
license: apache-2.0
tags:
- bert
- kcbert
- kor-hate-sentence
- sentimental-analysis
SJ-Donald/kor-hate-sentence-large
SJ-Donald/kor-hate-sentence-large is pretrained model using follow:
Models
Datasets
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}