ucirvine/sms_spam
Viewer • Updated • 5.57k • 7.27k • 58
How to use mariagrandury/roberta-base-finetuned-sms-spam-detection with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="mariagrandury/roberta-base-finetuned-sms-spam-detection") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("mariagrandury/roberta-base-finetuned-sms-spam-detection")
model = AutoModelForSequenceClassification.from_pretrained("mariagrandury/roberta-base-finetuned-sms-spam-detection", device_map="auto")This model is a fine-tuned version of roberta-base on the sms_spam 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:
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 0.0363 | 1.0 | 250 | 0.0156 | 0.996 |
| 0.0147 | 2.0 | 500 | 0.0133 | 0.998 |
Base model
FacebookAI/roberta-base