File size: 1,062 Bytes
cc9dda1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5fe23be
cc9dda1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9b7dc2c
cc9dda1
 
9b7dc2c
 
cc9dda1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
---
license: apache-2.0
tags:
- bert
- kcbert
- kor-hate-sentence
- sentimental-analysis
---

# SJ-Donald/kor-hate-sentence

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

## Models

* [beomi/kcbert-base](https://huggingface.co/beomi/kcbert-base)

## Datasets

* [SJ-Donald/kor-hate-sentence](https://huggingface.co/datasets/SJ-Donald/kor-hate-sentence)

## How to use

```Python
from transformers import TextClassificationPipeline, BertForSequenceClassification, AutoTokenizer+

model_name = 'SJ-Donald/kor-hate-sentence'
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.9767946600914001}
{'label': 'clean', 'score': 0.023970650508999825}
```