DataWizardd's picture
Update README.md
70420e5 verified
|
raw
history blame
1.31 kB
metadata
language: ko
datasets: naver-finance-news
tags:
  - sentiment-analysis
  - korean
  - finance
  - finbert
  - transformers
license: mit
model-index:
  - name: finbert-sentiment-ko
    results:
      - task:
          name: Sentiment Analysis
          type: text-classification
        metrics:
          - type: accuracy
            value: (์ง์ ‘ ์ธก์ •ํ•œ ์ •ํ™•๋„)

FinBERT Sentiment Analysis (Korean, Finance Domain)

์ด ๋ชจ๋ธ์€ ํ•œ๊ตญ์–ด ๊ธˆ์œต ๋‰ด์Šค ์š”์•ฝ๋ฌธ์„ ๋Œ€์ƒ์œผ๋กœ ๊ฐ์ •์„ ๋ถ„๋ฅ˜ํ•˜๊ธฐ ์œ„ํ•ด ํŒŒ์ธํŠœ๋‹๋œ BERT ๊ธฐ๋ฐ˜ ๋ชจ๋ธ์ž…๋‹ˆ๋‹ค.
๊ฐ์ • ๋ถ„๋ฅ˜๋Š” ๋‹ค์Œ ์„ธ ๊ฐ€์ง€ ํด๋ž˜์Šค ์ค‘ ํ•˜๋‚˜๋กœ ์ˆ˜ํ–‰๋ฉ๋‹ˆ๋‹ค:

  • 0: ๋ถ€์ •
  • 1: ์ค‘๋ฆฝ
  • 2: ๊ธ์ •

๐Ÿง  ํ•™์Šต ์ •๋ณด

  • ๊ธฐ๋ฐ˜ ๋ชจ๋ธ: snunlp/KR-FinBERT-SC
  • ๋ฐ์ดํ„ฐ: ์ง์ ‘ ์ˆ˜์ง‘ํ•œ ๋„ค์ด๋ฒ„ ๊ธˆ์œต ๋‰ด์Šค ์š”์•ฝ + ๊ฐ์ • ์ˆ˜์ž‘์—… ๋ผ๋ฒจ๋ง
  • ์ด ์ƒ˜ํ”Œ ์ˆ˜: ์•ฝ 200๊ฐœ
  • Optimizer: AdamW
  • Epochs: 4
  • ์ตœ๋Œ€ ๊ธธ์ด: 128
  • ํ‰๊ฐ€ ์ง€ํ‘œ: Accuracy, F1 Score

๐Ÿ›  ์‚ฌ์šฉ ๋ฐฉ๋ฒ•

from transformers import pipeline

pipe = pipeline("text-classification", model="DataWizardd/finbert-sentiment-ko")

# ์˜ˆ์‹œ
pipe("์›/๋‹ฌ๋Ÿฌ ํ™˜์œจ์ด ๊ธ‰๋“ฑํ•˜๋ฉฐ ์‹œ์žฅ ๋ถˆ์•ˆ์ด ์ปค์กŒ๋‹ค.")
# ์ถœ๋ ฅ: [{'label': 'LABEL_0', 'score': 0.95}] โ†’ ๋ถ€์ •