YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
SMS Spam Detection using DistilBERT
Model Description
This model is fine-tuned using DistilBERT for SMS spam classification.
Dataset
SMS Spam Collection Dataset
Labels
- 0: Ham (Normal message)
- 1: Spam
Model Architecture
DistilBERT transformer model fine-tuned for binary text classification.
Training
The model was trained using Google Colab with T4 GPU.
Evaluation Metrics
The model was evaluated using:
- Accuracy
- Precision
- Recall
- F1-Score
Intended Use
This model can classify SMS messages as spam or normal text.
- Downloads last month
- 4
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support