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}]
|
|