How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("mrm8488/bert-tiny-finetuned-sms-spam-detection")
model = AutoModelForSequenceClassification.from_pretrained("mrm8488/bert-tiny-finetuned-sms-spam-detection")
Quick Links

BERT-Tiny fine-tuned on on sms_spam dataset for spam detection

Validation accuray: 0.98

Downloads last month
54,207
Safetensors
Model size
4.39M params
Tensor type
I64
Β·
F32
Β·
Inference Providers NEW

Model tree for mrm8488/bert-tiny-finetuned-sms-spam-detection

Finetunes
2 models
Quantizations
1 model

Dataset used to train mrm8488/bert-tiny-finetuned-sms-spam-detection

Spaces using mrm8488/bert-tiny-finetuned-sms-spam-detection 45