File size: 2,655 Bytes
536e2d3
 
 
 
 
 
 
03d49e4
 
 
 
 
 
 
ffd1024
 
03d49e4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
f77aac8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
46
47
48
49
50
51
52
---
license: apache-2.0
language:
- ko
library_name: transformers
pipeline_tag: text-classification
---
# Usage
```python
# import library
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline

# load model
tokenizer = AutoTokenizer.from_pretrained("jaehyeong/koelectra-base-v3-generalized-sentiment-analysis")
model = AutoModelForSequenceClassification.from_pretrained("jaehyeong/koelectra-base-v3-generalized-sentiment-analysis")
sentiment_classifier = TextClassificationPipeline(tokenizer=tokenizer, model=model)

# target reviews
review_list = [
	'이쁘고 μ’‹μ•„μš”~~~씻기도 νŽΈν•˜κ³  아이고 μ΄μ˜λ‹€κ³  μžκΈ°λ°©μ— 갖닀놓고 μž˜μ¨μš”~^^',
	'아직 μž…μ–΄λ³΄μ§„ μ•Šμ•˜μ§€λ§Œ ꡉμž₯히 κ°€λ²Όμ›Œμš”~~ λ‹€λ₯Έ 리뷰처럼 어깑이 μ’€ λ˜λ„€μš”γ…‹ λ§Œμ‘±ν•©λ‹ˆλ‹€. μ—„μ²­ λΉ λ₯Έλ°œμ†‘ κ°μ‚¬λ“œλ €μš” :)',
	'재ꡬ맀 ν•œκ±΄λ° λ„ˆλ¬΄λ„ˆλ¬΄ 가성비인거 κ°™μ•„μš”!! λ‹€μŒμ— 또 μƒκ°λ‚˜λ©΄ 3개째 또 μ‚΄λ“―..γ…Žγ…Ž',
	'κ°€μŠ΅λŸ‰μ΄ λ„ˆλ¬΄ μ μ–΄μš”. 방이 μž‘μ§€ μ•Šλ‹€λ©΄ 무쑰건 ν°κ±Έλ‘œκ΅¬λ§€ν•˜μ„Έμš”. λ¬ΌλŸ‰λ„ μ‘°κΈˆλ°–μ— μ•ˆλ“€μ–΄κ°€μ„œ 쓰기도 λΆˆνŽΈν•¨',
	'ν•œλ²ˆμž…μ—ˆλŠ”λ° μ˜†μ— λ΄‰μ œμ„  λ‹€ 풀리고 μ‹€λ°₯도 계속 λ‚˜μ˜΅λ‹ˆλ‹€. 마감 처리 λ„ˆλ¬΄ 엉망 μ•„λ‹Œκ°€μš”?',
	'λ”°λœ»ν•˜κ³  μ’‹κΈ΄ν•œλ° 배솑이 λŠλ €μš”',
	'맛은 μžˆλŠ”λ° 가격이 μžˆλŠ” νŽΈμ΄μ—μš”'
]

# predict
for idx, review in enumerate(review_list):
  pred = sentiment_classifier(review)
  print(f'{review}\n>> {pred[0]}')
```
```
이쁘고 μ’‹μ•„μš”~~~씻기도 νŽΈν•˜κ³  아이고 μ΄μ˜λ‹€κ³  μžκΈ°λ°©μ— 갖닀놓고 μž˜μ¨μš”~^^
>> {'label': '1', 'score': 0.9945501685142517}
아직 μž…μ–΄λ³΄μ§„ μ•Šμ•˜μ§€λ§Œ ꡉμž₯히 κ°€λ²Όμ›Œμš”~~ λ‹€λ₯Έ 리뷰처럼 어깑이 μ’€ λ˜λ„€μš”γ…‹ λ§Œμ‘±ν•©λ‹ˆλ‹€. μ—„μ²­ λΉ λ₯Έλ°œμ†‘ κ°μ‚¬λ“œλ €μš” :)
>> {'label': '1', 'score': 0.995430588722229}
재ꡬ맀 ν•œκ±΄λ° λ„ˆλ¬΄λ„ˆλ¬΄ 가성비인거 κ°™μ•„μš”!! λ‹€μŒμ— 또 μƒκ°λ‚˜λ©΄ 3개째 또 μ‚΄λ“―..γ…Žγ…Ž
>> {'label': '1', 'score': 0.9959582686424255}
κ°€μŠ΅λŸ‰μ΄ λ„ˆλ¬΄ μ μ–΄μš”. 방이 μž‘μ§€ μ•Šλ‹€λ©΄ 무쑰건 ν°κ±Έλ‘œκ΅¬λ§€ν•˜μ„Έμš”. λ¬ΌλŸ‰λ„ μ‘°κΈˆλ°–μ— μ•ˆλ“€μ–΄κ°€μ„œ 쓰기도 λΆˆνŽΈν•¨
>> {'label': '0', 'score': 0.9984619617462158}
ν•œλ²ˆμž…μ—ˆλŠ”λ° μ˜†μ— λ΄‰μ œμ„  λ‹€ 풀리고 μ‹€λ°₯도 계속 λ‚˜μ˜΅λ‹ˆλ‹€. 마감 처리 λ„ˆλ¬΄ 엉망 μ•„λ‹Œκ°€μš”?
>> {'label': '0', 'score': 0.9991756677627563}
λ”°λœ»ν•˜κ³  μ’‹κΈ΄ν•œλ° 배솑이 λŠλ €μš”
>> {'label': '1', 'score': 0.6473883390426636}
맛은 μžˆλŠ”λ° 가격이 μžˆλŠ” νŽΈμ΄μ—μš”
>> {'label': '1', 'score': 0.5128092169761658}
```
- label 0 : negative review
- label 1 : positive review