You can deploy this model and start earning money today!

alizaabbas12/
distilbert-sms-spam-sargodha

Name: distilbert-sms-spam-sargodha
Availability: InStock

Views

No views yet

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

python

from transformers import pipeline
classifier = pipeline("text-classification", model="alizaabbas12/distilbert-sms-spam-sargodha")
classifier("Win free prize now")