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.