metadata
language: ko
datasets:
- investing_comments_krw
metrics:
- accuracy
- precision
- recall
- f1
tags:
- sentiment-analysis
- finance
- krw
- binary-classification
- emotion
model-index:
- name: FinBERT-Sentiment-KRW-Comment (v3)
results:
- task:
type: text-classification
name: Sentiment Analysis (Fear vs. Greed)
metrics:
- type: accuracy
value: 0.94
- type: precision
value: 0.94
- type: recall
value: 0.94
- type: f1
value: 0.94
FinBERT-Sentiment-KRW-Comment (v3)
์ด ๋ชจ๋ธ์ snunlp/KR-FinBERT-SC
๋ฅผ ๊ธฐ๋ฐ์ผ๋ก ํ์ธํ๋ํ ํ๊ตญ์ด ๊ธ์ต ๊ฐ์ ๋ถ์ ๋ชจ๋ธ์
๋๋ค.
ํนํ ํ์จ(FX) ๊ด๋ จ ๋๊ธ์์ ๋ํ๋๋ ๊ฐ์ ์ ๊ณตํฌ(0) ๋๋ ์์ฌ(1) ์ด์ง ๋ถ๋ฅํ๋ ๋ฐ ๋ชฉ์ ์ด ์์ต๋๋ค.
๐งพ ๋ผ๋ฒจ ์ ์
๋ผ๋ฒจ | ์ค๋ช |
---|---|
0 | ๊ณตํฌ (Fear) |
1 | ์์ฌ (Greed) |
๐๏ธโโ๏ธ ํ์ต ์ ๋ณด
- Base model:
snunlp/KR-FinBERT-SC
- Task: ๊ฐ์ ์ด์ง ๋ถ๋ฅ (๊ณตํฌ vs ์์ฌ)
- Input: ํ๊ตญ์ด ๋๊ธ (
content
) - Output: 0 ๋๋ 1
- Training epochs: 4
- Train size: ์ฝ
X
๊ฐ - Eval size: ์ฝ
X
๊ฐ - Evaluation metric: Accuracy, Precision, Recall, F1
๐ ํ๊ฐ ๊ฒฐ๊ณผ (Test Set ๊ธฐ์ค)
Metric | Score |
---|---|
Accuracy | 0.94 |
Precision | 0.94 |
Recall | 0.94 |
F1-score | 0.94 |
๊ณตํฌ(0): precision=0.95, recall=0.91, f1=0.93
์์ฌ(1): precision=0.93, recall=0.96, f1=0.94
๐งช ์ฌ์ฉ ์์
from transformers import pipeline
pipe = pipeline("text-classification", model="DataWizardd/finbert-sentiment-krw-comment-v3")
pipe("์ฌ์์น ์๋ค์ ์ค๋์ ์ ๋ฐ๋ฐํ๊ณ 1์8์ผ์ ๊น์ ์์ด ๋ฏธ์ฌ์ผ ์คํํ๊ณ ๊ทธ๋ฌ๋ฉด 1170์ ๊ธฐ๋ณธ์ด๊ณ 1190์๋ ์์๊ฐ์ผ๋ฏ.")
# โ [{'label': '1', 'score': 0.98}]