Downloads · 30 days
5
6% of all-time downloads
gpt2-spam-classifier
upster/gpt2-spam-classifier
gpt2-spam-classifier is a text classification model from upster. Use it when you need a label for a piece of text. The card lists the license as mit.
This model is a fine-tuned version of GPT-2 small for spam detection. It was trained on the SMS Spam Collection Dataset.
All-time downloads
89
Public
Repo size
1.1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin548 MB · 100%
At a glance
- Task
- Text Classification
- License
- mit
- Access
- Public
- Created
- Mar 15, 2025
- Updated
- Mar 15, 2025
- SHA
- d034d721
Try a prompt
Model card
From the Hugging Face model README
GPT-2 Spam Classifier
This model is a fine-tuned version of GPT-2 small for spam detection. It was trained on the SMS Spam Collection Dataset.
Model Details
- Base model: GPT-2 small (124M parameters)
- Task: Binary classification (spam vs. not spam)
- Training Data: SMS Spam Collection Dataset
- Fine-tuning approach: Last layer + classification head
Usage
The model expects text input and returns a binary classification (spam/not spam).
Performance
- Training accuracy: ~95%
- Validation accuracy: ~95%
- Test accuracy: ~93%
Limitations
This model was trained on SMS messages and may not generalize well to other types of text content.
- Task
- Text Classification
- License
- mit
- Languages
- en
- Created
- Mar 15, 2025
- Updated
- Mar 15, 2025