You can deploy this model and start earning money today!

Danisht/
spam-detector-distilbert

Name: spam-detector-distilbert
Availability: InStock

safetensors

distilbert

Views

No views yet

Spam Detector — DistilBERT

Model Description

A fine-tuned DistilBERT model for SMS spam detection. Classifies messages as SPAM or NOT SPAM.

Developed by: Muhammad Danisht
Model type: Text Classification
Language: English
Finetuned from: distilbert-base-uncased

Training Data

Dataset: SMS Spam Collection (5,574 messages)
Split: 80% train / 20% test

Training Results

Epoch	Accuracy
1	98.92%
2	98.92%
3	99.19%

How to Use

python

from transformers import pipeline
spam_detector = pipeline("text-classification", model="Danisht/spam-detector-distilbert")
result = spam_detector("Congratulations! You won a free iPhone!")
print(result)

Limitations

Model trained on 2011 SMS data. May miss modern phishing patterns.

GitHub

https://github.com/muhammeddanisht