Machine Learning
Collection
Just a little Machine Learning and Artificial Intelligence Projects with fine tuning and model editing • 3 items • Updated
How to use Umranz/Email-Spam-Classifier with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Umranz/Email-Spam-Classifier") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Umranz/Email-Spam-Classifier")
model = AutoModelForSequenceClassification.from_pretrained("Umranz/Email-Spam-Classifier")This model is a fine-tuned version of answerdotai/ModernBERT-base 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:
| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
|---|---|---|---|---|---|---|---|
| 0.0036 | 1.0 | 1250 | 0.0658 | 0.9895 | 0.9895 | 0.9895 | 0.9895 |
| 0.0191 | 2.0 | 2500 | 0.0915 | 0.9895 | 0.9895 | 0.9896 | 0.9895 |
| 0.0000 | 3.0 | 3750 | 0.0564 | 0.994 | 0.994 | 0.994 | 0.994 |
Base model
answerdotai/ModernBERT-base