Text Classification
Transformers
TensorBoard
Safetensors
distilbert
Generated from Trainer
text-embeddings-inference
Instructions to use SimoneJLaudani/test_trainer3 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SimoneJLaudani/test_trainer3 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="SimoneJLaudani/test_trainer3")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("SimoneJLaudani/test_trainer3") model = AutoModelForSequenceClassification.from_pretrained("SimoneJLaudani/test_trainer3", device_map="auto") - Notebooks
- Google Colab
- Kaggle
test_trainer3
This model is a fine-tuned version of distilbert-base-uncased on the None dataset. It achieves the following results on the evaluation set:
- Loss: 0.0615
- Precision: 1.0
- Recall: 1.0
- F1: 1.0
- Accuracy: 1.0
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 16
- eval_batch_size: 16
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 15
Training results
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|---|---|---|---|---|---|---|---|
| 1.9262 | 1.15 | 15 | 1.8025 | 0.4581 | 0.4524 | 0.3755 | 0.4524 |
| 1.6069 | 2.31 | 30 | 1.2284 | 1.0 | 1.0 | 1.0 | 1.0 |
| 1.0686 | 3.46 | 45 | 0.7174 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.6873 | 4.62 | 60 | 0.4283 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.4387 | 5.77 | 75 | 0.2676 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.2981 | 6.92 | 90 | 0.1794 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.202 | 8.08 | 105 | 0.1282 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.1588 | 9.23 | 120 | 0.0988 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.1276 | 10.38 | 135 | 0.0813 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.1058 | 11.54 | 150 | 0.0715 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.0951 | 12.69 | 165 | 0.0656 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.0906 | 13.85 | 180 | 0.0624 | 1.0 | 1.0 | 1.0 | 1.0 |
| 0.0886 | 15.0 | 195 | 0.0615 | 1.0 | 1.0 | 1.0 | 1.0 |
Framework versions
- Transformers 4.38.2
- Pytorch 2.1.0+cu121
- Datasets 2.18.0
- Tokenizers 0.15.2
- Downloads last month
- 6
Model tree for SimoneJLaudani/test_trainer3
Base model
distilbert/distilbert-base-uncased