Instructions to use SimoneJLaudani/trainer12 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SimoneJLaudani/trainer12 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="SimoneJLaudani/trainer12")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("SimoneJLaudani/trainer12") model = AutoModelForSequenceClassification.from_pretrained("SimoneJLaudani/trainer12", device_map="auto") - Notebooks
- Google Colab
- Kaggle
trainer12
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: 1.6394
- Precision: 0.5536
- Recall: 0.5476
- F1: 0.5366
- Accuracy: 0.5476
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: 8
- eval_batch_size: 8
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 10
Training results
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
|---|---|---|---|---|---|---|---|
| 0.2956 | 0.57 | 30 | 1.1151 | 0.6738 | 0.6429 | 0.6338 | 0.6429 |
| 0.2059 | 1.13 | 60 | 1.1796 | 0.5830 | 0.5714 | 0.5522 | 0.5714 |
| 0.1118 | 1.7 | 90 | 1.2686 | 0.5442 | 0.5357 | 0.5200 | 0.5357 |
| 0.062 | 2.26 | 120 | 1.3631 | 0.5856 | 0.5595 | 0.5430 | 0.5595 |
| 0.0402 | 2.83 | 150 | 1.4179 | 0.5795 | 0.5714 | 0.5525 | 0.5714 |
| 0.0206 | 3.4 | 180 | 1.4027 | 0.6128 | 0.6071 | 0.5928 | 0.6071 |
| 0.0173 | 3.96 | 210 | 1.4487 | 0.6309 | 0.6310 | 0.6231 | 0.6310 |
| 0.0122 | 4.53 | 240 | 1.5266 | 0.5927 | 0.5833 | 0.5674 | 0.5833 |
| 0.0107 | 5.09 | 270 | 1.4902 | 0.6317 | 0.6310 | 0.6247 | 0.6310 |
| 0.0084 | 5.66 | 300 | 1.5660 | 0.5715 | 0.5595 | 0.5488 | 0.5595 |
| 0.0075 | 6.23 | 330 | 1.5799 | 0.5830 | 0.5714 | 0.5709 | 0.5714 |
| 0.0066 | 6.79 | 360 | 1.6132 | 0.5517 | 0.5476 | 0.5360 | 0.5476 |
| 0.0059 | 7.36 | 390 | 1.6275 | 0.5340 | 0.5238 | 0.5144 | 0.5238 |
| 0.0054 | 7.92 | 420 | 1.6335 | 0.5721 | 0.5595 | 0.5579 | 0.5595 |
| 0.0055 | 8.49 | 450 | 1.6374 | 0.5427 | 0.5357 | 0.5239 | 0.5357 |
| 0.0051 | 9.06 | 480 | 1.6348 | 0.5673 | 0.5595 | 0.5511 | 0.5595 |
| 0.005 | 9.62 | 510 | 1.6391 | 0.5536 | 0.5476 | 0.5366 | 0.5476 |
Framework versions
- Transformers 4.38.2
- Pytorch 2.2.1+cu121
- Datasets 2.18.0
- Tokenizers 0.15.2
- Downloads last month
- 3
Model tree for SimoneJLaudani/trainer12
Base model
distilbert/distilbert-base-uncased