File size: 751 Bytes
97ac59c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
---
language: en
tags:
- spam-detection
- text-classification
- gpt2
license: mit
---
# 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.
|