Hugging Face's logo

luluw
/
distilbert-base-uncased-finetuned-sentiment

Text Classification
Transformers
TensorBoard
Safetensors
English
distilbert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
luluw commited on
Commit
a3094cf
•
1 Parent(s): 418f1cb

End of training

Browse files
Files changed (4) hide show
  1. README.md +8 -8
  2. model.safetensors +1 -1
  3. runs/Nov11_13-13-40_d4db5581afa5/events.out.tfevents.1731330840.d4db5581afa5.368.0 +2 -2
  4. runs/Nov11_13-13-40_d4db5581afa5/events.out.tfevents.1731334095.d4db5581afa5.368.1 +3 -0
README.md CHANGED
@@ -20,8 +20,8 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb-dataset-of-50k-movie-reviews dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.2166
24
- - Accuracy: 0.9263
25
 
26
  ## Model description
27
 
@@ -40,7 +40,7 @@ More information needed
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
- - learning_rate: 2.5e-05
44
  - train_batch_size: 32
45
  - eval_batch_size: 32
46
  - seed: 42
@@ -54,11 +54,11 @@ The following hyperparameters were used during training:
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
56
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
57
- | 0.3029 | 1.0 | 1250 | 0.2369 | 0.9085 |
58
- | 0.1607 | 2.0 | 2500 | 0.2166 | 0.9263 |
59
- | 0.0924 | 3.0 | 3750 | 0.2867 | 0.9208 |
60
- | 0.0521 | 4.0 | 5000 | 0.3193 | 0.9235 |
61
- | 0.0312 | 5.0 | 6250 | 0.3764 | 0.9227 |
62
 
63
 
64
  ### Framework versions
 
20
 
21
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb-dataset-of-50k-movie-reviews dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.3021
24
+ - Accuracy: 0.9144
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - learning_rate: 2e-05
44
  - train_batch_size: 32
45
  - eval_batch_size: 32
46
  - seed: 42
 
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
56
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
57
+ | 0.3378 | 1.0 | 1250 | 0.2327 | 0.9042 |
58
+ | 0.186 | 2.0 | 2500 | 0.2519 | 0.9117 |
59
+ | 0.1135 | 3.0 | 3750 | 0.3021 | 0.9144 |
60
+ | 0.0706 | 4.0 | 5000 | 0.3474 | 0.9125 |
61
+ | 0.0453 | 5.0 | 6250 | 0.3919 | 0.9135 |
62
 
63
 
64
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed6620db1e433cd1f87f379986671ddc3eefb41c0442f365a86368d86c667a8c
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb48832e7ac085ac23ff1bc3aff58eeda4058dc817324e6653a4f9fcfa0fa040
3
  size 267832560
runs/Nov11_13-13-40_d4db5581afa5/events.out.tfevents.1731330840.d4db5581afa5.368.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3e933995ed677b3f289a40fd86d926d378bda3966d104896a6f4d46b2013ca07
3
- size 7347
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a352d4a5d551b452555386001c8ceb51c6dbe4a6ebf00e4f0982dab0cf82255e
3
+ size 8024
runs/Nov11_13-13-40_d4db5581afa5/events.out.tfevents.1731334095.d4db5581afa5.368.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5261c756881957b325f9a99c64b7a754ded535794c01c4cc3154a88807981d5c
3
+ size 411