Hugging Face's logo

sandromck
/
movie-genre-predictions

Text Classification
Transformers
PyTorch
bert
Generated from Trainer
Inference Endpoints
Model card Files Community
1
sandromck commited on
Commit
34113b1
•
1 Parent(s): 2a25eb5

End of training

Browse files
Files changed (1) hide show
  1. README.md +24 -14
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 1.6582
21
- - Accuracy: 0.4465
22
 
23
  ## Model description
24
 
@@ -38,23 +38,33 @@ More information needed
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 1e-05
41
- - train_batch_size: 32
42
- - eval_batch_size: 64
43
- - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
- - num_epochs: 6
47
 
48
  ### Training results
49
 
50
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
- | 1.6522 | 1.0 | 1159 | 1.5895 | 0.4441 |
53
- | 1.5335 | 2.0 | 2318 | 1.5680 | 0.4526 |
54
- | 1.3944 | 3.0 | 3477 | 1.5825 | 0.4498 |
55
- | 1.2903 | 4.0 | 4636 | 1.6169 | 0.4485 |
56
- | 1.205 | 5.0 | 5795 | 1.6431 | 0.4471 |
57
- | 1.1465 | 6.0 | 6954 | 1.6582 | 0.4465 |
 
 
 
 
 
 
 
 
 
 
58
 
59
 
60
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 2.9264
21
+ - Accuracy: 0.3289
22
 
23
  ## Model description
24
 
 
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 1e-05
41
+ - train_batch_size: 16
42
+ - eval_batch_size: 16
43
+ - seed: 67
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
+ - num_epochs: 16
47
 
48
  ### Training results
49
 
50
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|
52
+ | 2.1244 | 1.0 | 2318 | 2.0165 | 0.2812 |
53
+ | 1.8772 | 2.0 | 4636 | 1.8726 | 0.3393 |
54
+ | 1.7305 | 3.0 | 6954 | 1.8347 | 0.3589 |
55
+ | 1.5873 | 4.0 | 9272 | 1.8587 | 0.3631 |
56
+ | 1.4615 | 5.0 | 11590 | 1.9537 | 0.3588 |
57
+ | 1.2914 | 6.0 | 13908 | 2.0050 | 0.3620 |
58
+ | 1.15 | 7.0 | 16226 | 2.1050 | 0.3467 |
59
+ | 1.0332 | 8.0 | 18544 | 2.2455 | 0.3490 |
60
+ | 0.9257 | 9.0 | 20862 | 2.3592 | 0.3357 |
61
+ | 0.8086 | 10.0 | 23180 | 2.4679 | 0.3343 |
62
+ | 0.7231 | 11.0 | 25498 | 2.5798 | 0.3362 |
63
+ | 0.6471 | 12.0 | 27816 | 2.6849 | 0.3335 |
64
+ | 0.5839 | 13.0 | 30134 | 2.7941 | 0.3256 |
65
+ | 0.5238 | 14.0 | 32452 | 2.8416 | 0.3256 |
66
+ | 0.4804 | 15.0 | 34770 | 2.9133 | 0.3297 |
67
+ | 0.4346 | 16.0 | 37088 | 2.9264 | 0.3289 |
68
 
69
 
70
  ### Framework versions