LimYeri commited on
Commit
1ffbc3e
·
verified ·
1 Parent(s): bcfe4ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -0
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