rmsdud/kobert-classifier

KoBERT 분류 모델

이 모델은 KoBERT를 기반으로 텍스트 분류를 위해 파인튜닝된 모델입니다.

모델 정보

• 기본 모델: beomi/kcbert-base
• 클래스 수: 12
• 사용 방법: 아래 코드를 참조하세요

사용 예시

from transformers import BertForSequenceClassification, BertTokenizer

# 모델과 토크나이저 로드
model_name = "rmsdud/kobert-classifier"
tokenizer = BertTokenizer.from_pretrained(model_name)
model = BertForSequenceClassification.from_pretrained(model_name)

# 추론
text = "분류할 텍스트를 입력하세요."
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128)
outputs = model(**inputs)
logits = outputs.logits
predicted_class = logits.argmax(-1).item()
print(f"예측 클래스: predicted_class")