pgos-emotion-stage1-major

PGOS Emotion Agent — Stage 1 (대분류)

Psychological Growth OS(PGOS) 프로젝트의 감정 분류 모델입니다. 계층적 2단계 분류기의 1단계 대분류 모델입니다.

모델 개요

입력 텍스트를 6개 감정 대분류로 분류합니다.

성능

학습 데이터

• 원천: AI Hub 018 감성대화 말뭉치
• 데이터: 클래스당 균등 샘플링 + GPT-4o-mini 증강
• 총 학습 데이터: 64,000개

베이스 모델

beomi/KcELECTRA-base-v2022

사용 방법

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

tokenizer = AutoTokenizer.from_pretrained("nuguri01/pgos-emotion-stage1-major")
model = AutoModelForSequenceClassification.from_pretrained("nuguri01/pgos-emotion-stage1-major")

text = "오늘 너무 화가 나고 짜증이 밀려와요"
inputs = tokenizer(text, return_tensors="pt", max_length=128, truncation=True)
with torch.no_grad():
    logits = model(**inputs).logits
pred = torch.argmax(logits, dim=-1).item()
labels = ["분노계", "슬픔계", "불안계", "상처계", "당황계", "기쁨계"]
print(f"대분류: {labels[pred]}")

PGOS 파이프라인

입력 텍스트 ↓ Stage 1 (이 모델) → 대분류 예측 ↓ Stage 2 (소분류 모델) → 세부 감정 예측 ↓ Top-3 감정 반환 (Hit Rate 0.877)

관련 모델

• Stage 2: pgos-emotion-stage2-anger
• Stage 2: pgos-emotion-stage2-sadness
• Stage 2: pgos-emotion-stage2-anxiety
• Stage 2: pgos-emotion-stage2-hurt
• Stage 2: pgos-emotion-stage2-embarrass
• Stage 2: pgos-emotion-stage2-joy

프로젝트

• 서비스: Psychological Growth OS (PGOS)
• 철학: Carl Rogers 인간중심치료 + ACT Hexaflex
• Behavior Agent: nuguri01/pgos-behavior-agent