sentiment-analysis-roberta-base-V1.4_ima_ds
This model is a fine-tuned version of FacebookAI/roberta-base on the None dataset. It achieves the following results on the evaluation set:
- Loss: 1.3065
- Accuracy: 0.6045
- Precision Macro: 0.5769
- Recall Macro: 0.5921
- F1 Macro: 0.5726
- Precision Weighted: 0.6266
- Recall Weighted: 0.6045
- F1 Weighted: 0.6049
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: 32
- eval_batch_size: 8
- seed: 3407
- optimizer: Use OptimizerNames.ADAMW_8BIT with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: cosine
- lr_scheduler_warmup_steps: 50
- num_epochs: 20
Training results
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision Macro | Recall Macro | F1 Macro | Precision Weighted | Recall Weighted | F1 Weighted |
|---|---|---|---|---|---|---|---|---|---|---|
| 1.5869 | 0.4 | 20 | 1.6105 | 0.2821 | 0.0564 | 0.2 | 0.0880 | 0.0796 | 0.2821 | 0.1242 |
| 1.6423 | 0.8 | 40 | 1.5978 | 0.2922 | 0.2473 | 0.2177 | 0.1478 | 0.3269 | 0.2922 | 0.1993 |
| 1.4869 | 1.2 | 60 | 1.4559 | 0.4383 | 0.3448 | 0.3580 | 0.3347 | 0.4135 | 0.4383 | 0.4073 |
| 1.4004 | 1.6 | 80 | 1.3144 | 0.5139 | 0.4732 | 0.4553 | 0.4486 | 0.5264 | 0.5139 | 0.5014 |
| 1.2658 | 2.0 | 100 | 1.3022 | 0.4710 | 0.4792 | 0.4561 | 0.4008 | 0.5682 | 0.4710 | 0.4429 |
| 1.3302 | 2.4 | 120 | 1.2608 | 0.4861 | 0.5356 | 0.5254 | 0.4649 | 0.5991 | 0.4861 | 0.4965 |
| 0.8278 | 2.8 | 140 | 1.2198 | 0.5516 | 0.5583 | 0.5455 | 0.5293 | 0.5972 | 0.5516 | 0.5579 |
| 0.8199 | 3.2 | 160 | 1.1870 | 0.5743 | 0.5562 | 0.5561 | 0.5367 | 0.6063 | 0.5743 | 0.5728 |
| 0.9208 | 3.6 | 180 | 1.1146 | 0.5693 | 0.5529 | 0.5727 | 0.5439 | 0.6211 | 0.5693 | 0.5805 |
| 0.3989 | 4.0 | 200 | 1.1472 | 0.5693 | 0.5512 | 0.5756 | 0.5466 | 0.6132 | 0.5693 | 0.5819 |
| 0.8179 | 4.4 | 220 | 1.1235 | 0.6045 | 0.5704 | 0.5953 | 0.5772 | 0.6201 | 0.6045 | 0.6086 |
| 0.8518 | 4.8 | 240 | 1.2760 | 0.5869 | 0.5774 | 0.5818 | 0.5599 | 0.6278 | 0.5869 | 0.5955 |
| 0.641 | 5.2 | 260 | 1.2151 | 0.5668 | 0.5487 | 0.5737 | 0.5461 | 0.6013 | 0.5668 | 0.5767 |
| 0.4873 | 5.6 | 280 | 1.3065 | 0.6045 | 0.5769 | 0.5921 | 0.5726 | 0.6266 | 0.6045 | 0.6049 |
Framework versions
- Transformers 4.51.3
- Pytorch 2.6.0+cu124
- Datasets 3.6.0
- Tokenizers 0.21.1
- Downloads last month
- 4
Model tree for IMA-StreamSolve/sentiment-analysis-roberta-base-V1.4_ima_ds
Base model
FacebookAI/roberta-base