Views
No views yet
korean-emotion-kluebert-v2은 klue/bert-base를 기반으로 파인튜닝된 모델입니다.| Label ID | 감정 라벨 | 설명 |
|---|---|---|
| 0 | 공포 | 두려움이나 위협을 느끼는 감정 |
| 1 | 놀람 | 예상치 못한 상황에 대한 반응 |
| 2 | 분노 | 화나고 짜증나는 감정 |
| 3 | 슬픔 | 우울하고 침울한 감정 |
| 4 | 중립 | 특별한 감정 없이 평범한 상태 |
| 5 | 행복 | 기쁘고 즐거운 감정 |
| 6 | 혐오 | 불쾌하거나 싫어하는 감정 |
1from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
2
3# Load tokenizer and model
4tokenizer = AutoTokenizer.from_pretrained("dlckdfuf141/korean-emotion-kluebert-v2")
5model = AutoModelForSequenceClassification.from_pretrained("dlckdfuf141/korean-emotion-kluebert-v2")
6
7# Create pipeline
8emotion_classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
9
10# Predict emotion
11text = "오늘 하루 너무 즐거웠어!"
12result = emotion_classifier(text)
13print(result)klue/bert-base1[
2 {
3 "label": "행복",
4 "score": 0.9831
5 }
6]