Update README.md
Browse files
README.md
CHANGED
|
@@ -16,12 +16,12 @@ model-index:
|
|
| 16 |
type: text-classification
|
| 17 |
metrics:
|
| 18 |
- type: accuracy
|
| 19 |
-
value:
|
| 20 |
---
|
| 21 |
|
| 22 |
# FinBERT Sentiment Analysis (Korean, Finance Domain)
|
| 23 |
|
| 24 |
-
μ΄ λͺ¨λΈμ **νκ΅μ΄ κΈμ΅ λ΄μ€ μμ½λ¬Έ**μ λμμΌλ‘ κ°μ μ λΆλ₯νκΈ° μν΄ νμΈνλλ BERT κΈ°λ° λͺ¨λΈμ
λλ€.
|
| 25 |
κ°μ λΆλ₯λ λ€μ μΈ κ°μ§ ν΄λμ€ μ€ νλλ‘ μνλ©λλ€:
|
| 26 |
|
| 27 |
- `0`: λΆμ
|
|
@@ -31,13 +31,27 @@ model-index:
|
|
| 31 |
## π§ νμ΅ μ 보
|
| 32 |
|
| 33 |
- κΈ°λ° λͺ¨λΈ: [`snunlp/KR-FinBERT-SC`](https://huggingface.co/snunlp/KR-FinBERT-SC)
|
| 34 |
-
- λ°μ΄ν°: μ§μ μμ§ν **λ€μ΄λ² κΈμ΅ λ΄μ€** μμ½ + κ°μ μμμ
λΌλ²¨λ§
|
| 35 |
-
- μ΄ μν μ: μ½ 200
|
| 36 |
- Optimizer: AdamW
|
| 37 |
- Epochs: 4
|
| 38 |
- μ΅λ κΈΈμ΄: 128
|
| 39 |
- νκ° μ§ν: Accuracy, F1 Score
|
| 40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
## π μ¬μ© λ°©λ²
|
| 42 |
|
| 43 |
```python
|
|
@@ -45,6 +59,5 @@ from transformers import pipeline
|
|
| 45 |
|
| 46 |
pipe = pipeline("text-classification", model="DataWizardd/finbert-sentiment-ko")
|
| 47 |
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
# μΆλ ₯: [{'label': 'LABEL_0', 'score': 0.95}] β λΆμ
|
|
|
|
| 16 |
type: text-classification
|
| 17 |
metrics:
|
| 18 |
- type: accuracy
|
| 19 |
+
value: 0.93
|
| 20 |
---
|
| 21 |
|
| 22 |
# FinBERT Sentiment Analysis (Korean, Finance Domain)
|
| 23 |
|
| 24 |
+
μ΄ λͺ¨λΈμ **νκ΅μ΄ νμ¨(κΈμ΅) λ΄μ€ μμ½λ¬Έ**μ λμμΌλ‘ κ°μ μ λΆλ₯νκΈ° μν΄ νμΈνλλ BERT κΈ°λ° λͺ¨λΈμ
λλ€.
|
| 25 |
κ°μ λΆλ₯λ λ€μ μΈ κ°μ§ ν΄λμ€ μ€ νλλ‘ μνλ©λλ€:
|
| 26 |
|
| 27 |
- `0`: λΆμ
|
|
|
|
| 31 |
## π§ νμ΅ μ 보
|
| 32 |
|
| 33 |
- κΈ°λ° λͺ¨λΈ: [`snunlp/KR-FinBERT-SC`](https://huggingface.co/snunlp/KR-FinBERT-SC)
|
| 34 |
+
- λ°μ΄ν°: μ§μ μμ§ν **λ€μ΄λ² νμ¨(κΈμ΅) λ΄μ€** μμ½ + κ°μ μμμ
λΌλ²¨λ§
|
| 35 |
+
- μ΄ μν μ: μ½ 200
|
| 36 |
- Optimizer: AdamW
|
| 37 |
- Epochs: 4
|
| 38 |
- μ΅λ κΈΈμ΄: 128
|
| 39 |
- νκ° μ§ν: Accuracy, F1 Score
|
| 40 |
|
| 41 |
+
## π μ±λ₯ νκ°
|
| 42 |
+
|
| 43 |
+
| κ°μ ν΄λμ€ | Precision | Recall | F1-score | Support |
|
| 44 |
+
|-------------|-----------|--------|----------|---------|
|
| 45 |
+
| λΆμ | 0.89 | 1.00 | 0.94 | 17 |
|
| 46 |
+
| μ€λ¦½ | 1.00 | 0.82 | 0.90 | 11 |
|
| 47 |
+
| κΈμ | 0.93 | 0.93 | 0.93 | 14 |
|
| 48 |
+
| **μ νλ** | | | **0.93** | 42 |
|
| 49 |
+
|
| 50 |
+
> μ 체 μ νλ: **93%**
|
| 51 |
+
> Macro F1-score: **0.92**
|
| 52 |
+
|
| 53 |
+
---
|
| 54 |
+
|
| 55 |
## π μ¬μ© λ°©λ²
|
| 56 |
|
| 57 |
```python
|
|
|
|
| 59 |
|
| 60 |
pipe = pipeline("text-classification", model="DataWizardd/finbert-sentiment-ko")
|
| 61 |
|
| 62 |
+
pipe("νμ¨μ΄ κΈλ±νλ©° μμ₯ λΆμμ΄ μ»€μ§κ³ μλ€.")
|
| 63 |
+
# μΆλ ₯: [{'label': 'λΆμ ', 'score': 0.95}]
|
|
|