siddharthsingh5010 commited on
Commit
8d05231
1 Parent(s): 2e92ec0

Training completed!

Browse files
README.md CHANGED
@@ -23,10 +23,10 @@ model-index:
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
- value: 0.9235
27
  - name: F1
28
  type: f1
29
- value: 0.9231534611856926
30
  ---
31
 
32
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -36,9 +36,9 @@ should probably proofread and complete it, then remove this comment. -->
36
 
37
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
38
  It achieves the following results on the evaluation set:
39
- - Loss: 0.2196
40
- - Accuracy: 0.9235
41
- - F1: 0.9232
42
 
43
  ## Model description
44
 
@@ -58,8 +58,8 @@ More information needed
58
 
59
  The following hyperparameters were used during training:
60
  - learning_rate: 2e-05
61
- - train_batch_size: 64
62
- - eval_batch_size: 64
63
  - seed: 42
64
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
65
  - lr_scheduler_type: linear
@@ -69,13 +69,13 @@ The following hyperparameters were used during training:
69
 
70
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
71
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
72
- | 0.8302 | 1.0 | 250 | 0.3188 | 0.901 | 0.9002 |
73
- | 0.2473 | 2.0 | 500 | 0.2196 | 0.9235 | 0.9232 |
74
 
75
 
76
  ### Framework versions
77
 
78
- - Transformers 4.41.1
79
- - Pytorch 2.3.0+cu121
80
- - Datasets 2.19.1
81
- - Tokenizers 0.19.1
 
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
+ value: 0.8945
27
  - name: F1
28
  type: f1
29
+ value: 0.8905629739592735
30
  ---
31
 
32
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
36
 
37
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
38
  It achieves the following results on the evaluation set:
39
+ - Loss: 0.3513
40
+ - Accuracy: 0.8945
41
+ - F1: 0.8906
42
 
43
  ## Model description
44
 
 
58
 
59
  The following hyperparameters were used during training:
60
  - learning_rate: 2e-05
61
+ - train_batch_size: 128
62
+ - eval_batch_size: 128
63
  - seed: 42
64
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
65
  - lr_scheduler_type: linear
 
69
 
70
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
71
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
72
+ | No log | 1.0 | 125 | 0.5315 | 0.8495 | 0.8357 |
73
+ | 0.7527 | 2.0 | 250 | 0.3513 | 0.8945 | 0.8906 |
74
 
75
 
76
  ### Framework versions
77
 
78
+ - Transformers 4.39.3
79
+ - Pytorch 2.1.2
80
+ - Datasets 2.18.0
81
+ - Tokenizers 0.15.2
config.json CHANGED
@@ -36,6 +36,6 @@
36
  "sinusoidal_pos_embds": false,
37
  "tie_weights_": true,
38
  "torch_dtype": "float32",
39
- "transformers_version": "4.41.1",
40
  "vocab_size": 30522
41
  }
 
36
  "sinusoidal_pos_embds": false,
37
  "tie_weights_": true,
38
  "torch_dtype": "float32",
39
+ "transformers_version": "4.39.3",
40
  "vocab_size": 30522
41
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:341b7369ba76990031c29a19cbfd7e6fe446733ec68232fecc759948c09bd826
3
  size 267844872
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c4d7b36a45b1af67627159791a0c9e22cf01666303f9d70396ab3e0a6bcf626
3
  size 267844872
runs/May28_20-36-27_35cb10cdf83e/events.out.tfevents.1716928596.35cb10cdf83e.34.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7634313e24b5ed2d31b00a1ca8f3a1557a55c7076023b3f8067dafdbc83eb317
3
+ size 4832
runs/May28_20-38-07_35cb10cdf83e/events.out.tfevents.1716928690.35cb10cdf83e.34.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3def954adea668f865a677d60eb22e5b89f30d35b3fafb7238e901403d9bc312
3
+ size 4832
runs/May28_20-38-07_35cb10cdf83e/events.out.tfevents.1716928817.35cb10cdf83e.34.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38d7364ed7b09a848a2ebc0b94fefdd7892664940f7777692c7b77dfe7ea6de6
3
+ size 4184
runs/May28_20-43-14_35cb10cdf83e/events.out.tfevents.1716928997.35cb10cdf83e.109.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:476fbaca18a3220cde558dff0672d03caded01ad0f17737aaa211eb79ed5248e
3
+ size 6128
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cd3dec52eeb9b9734345e318a44d40f7680ac02b95c285a7e857344218836b70
3
- size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bdb7f063289e7f7ff608938857dccc7e8405f2c89038e2c1735aa4ea882726b0
3
+ size 4984