SJ-Donald/kcbert-large-unsmile

SJ-Donald/kcbert-large-unsmile is pretrained model using follow:

Models

• beomi/kcbert-large

Datasets

• smilegate-ai/kor_unsmile

How to use

from transformers import TextClassificationPipeline, BertForSequenceClassification, AutoTokenizer+

model_name = 'SJ-Donald/kcbert-large-unsmile'
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': '여성/가족', 'score': 0.9793611168861389}
{'label': '남성', 'score': 0.006330598145723343}
{'label': '성소수자', 'score': 0.007870828732848167}
{'label': '인종/국적', 'score': 0.010810344479978085}
{'label': '연령', 'score': 0.020540334284305573}
{'label': '지역', 'score': 0.015790466219186783}
{'label': '종교', 'score': 0.014563685283064842}
{'label': '기타 혐오', 'score': 0.04097242280840874}
{'label': '악플/욕설', 'score': 0.019168635830283165}
{'label': 'clean', 'score': 0.014866289682686329}