blockenters/sms-spam-classifier

SMS 스팸 분류기

학습에 사용된 데이터를 한글 SMS를 직접 가공하여 만들었습니다. 데이터셋이 궁금하시면, 문의 주세요.

이 모델은 SMS 스팸 탐지를 위해 미세 조정된 BERT 기반 다국어 모델입니다. SMS 메시지를 ham(비스팸) 또는 **spam(스팸)**으로 분류할 수 있습니다. Hugging Face Transformers 라이브러리의 bert-base-multilingual-cased 모델을 기반으로 학습되었습니다.

---

모델 세부정보

• 기본 모델: bert-base-multilingual-cased
• 태스크: 문장 분류(Sequence Classification)
• 지원 언어: 다국어
• 라벨 수: 2 (ham, spam)
• 데이터셋: 클린된 SMS 스팸 데이터셋

---

데이터셋 정보

훈련 및 평가에 사용된 데이터셋은 ham(비스팸) 또는 spam(스팸)으로 라벨링된 SMS 메시지를 포함하고 있습니다. 데이터는 전처리를 거친 후 다음과 같이 분리되었습니다:

• 훈련 데이터: 80%
• 검증 데이터: 20%

---

학습 설정

• 학습률(Learning Rate): 2e-5
• 배치 크기(Batch Size): 8 (디바이스 당)
• 에포크(Epochs): 1
• 평가 전략: 에포크 단위
• 토크나이저: bert-base-multilingual-cased

이 모델은 Hugging Face의 Trainer API를 사용하여 효율적으로 미세 조정되었습니다.

---

사용 방법

이 모델은 Hugging Face Transformers 라이브러리를 통해 바로 사용할 수 있습니다:

from transformers import AutoTokenizer, AutoModelForSequenceClassification

# 모델과 토크나이저 로드
tokenizer = AutoTokenizer.from_pretrained("blockenters/sms-spam-classifier")
model = AutoModelForSequenceClassification.from_pretrained("blockenters/sms-spam-classifier")

# 입력 샘플
text = "축하합니다! 무료 발리 여행 티켓을 받으셨습니다. WIN이라고 회신하세요."

# 토큰화 및 예측
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128)
outputs = model(**inputs)
predictions = outputs.logits.argmax(dim=-1)

# 예측 결과 디코딩
label_map = {0: "ham", 1: "spam"}
print(f"예측 결과: {label_map[predictions.item()]}")