| 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. | |