SMS Spam Detection with DistilBERT
Model Description
This model classifies SMS messages as Spam or Ham using DistilBERT.
Dataset
SMS Spam Collection Dataset from Kaggle
Training Details
- Model: distilbert-base-uncased
- Epochs: 2
- Batch Size: 16
- Trained on: Google Colab GPU
Results
- Accuracy: 99.19%
- Precision: 100%
- Recall: 94.12%
- F1-Score: 96.91%
How to use
from transformers import pipeline
classifier = pipeline("text-classification", model="alizaabbas12/distilbert-sms-spam-sargodha")
classifier("Win free prize now")