Update README.md
Browse files
README.md
CHANGED
|
@@ -116,6 +116,32 @@ result = predict_emotion("오늘 정말 기분이 좋고 행복한 하루였어!
|
|
| 116 |
print(result)
|
| 117 |
```
|
| 118 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 119 |
---
|
| 120 |
|
| 121 |
## Training Details
|
|
|
|
| 116 |
print(result)
|
| 117 |
```
|
| 118 |
|
| 119 |
+
### pipeline
|
| 120 |
+
```python
|
| 121 |
+
from transformers import pipeline
|
| 122 |
+
|
| 123 |
+
model_name = "LimYeri/HowRU-KoELECTRA-Emotion-Classifier"
|
| 124 |
+
|
| 125 |
+
classifier = pipeline(
|
| 126 |
+
"text-classification",
|
| 127 |
+
model=model_name,
|
| 128 |
+
tokenizer=model_name,
|
| 129 |
+
top_k=None # 전체 감정 확률 반환
|
| 130 |
+
)
|
| 131 |
+
|
| 132 |
+
# 예측
|
| 133 |
+
text = "오늘 정말 기분이 좋고 행복한 하루였어!"
|
| 134 |
+
result = classifier(text)
|
| 135 |
+
|
| 136 |
+
result = result[0]
|
| 137 |
+
|
| 138 |
+
print("입력 문장:", text)
|
| 139 |
+
print("\nTop-1 감정:", result[0]['label'], f"({result[0]['score']:.4f})")
|
| 140 |
+
print("\n전체 감정 분포:")
|
| 141 |
+
for r in result:
|
| 142 |
+
print(f" {r['label']}: {r['score']:.4f}")
|
| 143 |
+
```
|
| 144 |
+
|
| 145 |
---
|
| 146 |
|
| 147 |
## Training Details
|