ucirvine/sms_spam
Viewer • Updated • 5.57k • 5.23k • 55
How to use ZachBeesley/Spam-Detector with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="ZachBeesley/Spam-Detector") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("ZachBeesley/Spam-Detector")
model = AutoModelForSequenceClassification.from_pretrained("ZachBeesley/Spam-Detector")This model is a fine-tuned version of distilbert-base-uncased on an unknown dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Train Loss | Epoch |
|---|---|
| 0.0644 | 0 |
| 0.0209 | 1 |
| 0.0093 | 2 |
Base model
distilbert/distilbert-base-uncased