adith-ds commited on
Commit
20e363d
·
verified ·
1 Parent(s): b52fa9e

adith-ds/emotion-classifier-v2

Browse files
Files changed (3) hide show
  1. README.md +15 -8
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -16,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.3234
20
- - Macro F1: 0.7364
21
- - Macro Precision: 0.8012
22
- - Macro Recall: 0.6946
23
  - Macro Support: None
24
 
25
  ## Model description
@@ -46,15 +46,22 @@ The following hyperparameters were used during training:
46
  - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 500
49
- - num_epochs: 3
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Macro F1 | Macro Precision | Macro Recall | Macro Support |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------------:|:------------:|:-------------:|
55
- | No log | 1.0 | 86 | 0.5865 | 0.1447 | 0.1133 | 0.2 | None |
56
- | 0.6401 | 2.0 | 172 | 0.3973 | 0.6073 | 0.7713 | 0.5602 | None |
57
- | 0.4447 | 3.0 | 258 | 0.3234 | 0.7364 | 0.8012 | 0.6946 | None |
 
 
 
 
 
 
 
58
 
59
 
60
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.3137
20
+ - Macro F1: 0.8277
21
+ - Macro Precision: 0.8600
22
+ - Macro Recall: 0.8001
23
  - Macro Support: None
24
 
25
  ## Model description
 
46
  - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 500
49
+ - num_epochs: 10
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Macro F1 | Macro Precision | Macro Recall | Macro Support |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------------:|:------------:|:-------------:|
55
+ | No log | 1.0 | 86 | 0.5569 | 0.1457 | 0.1717 | 0.1943 | None |
56
+ | 0.6269 | 2.0 | 172 | 0.3887 | 0.6303 | 0.7830 | 0.5848 | None |
57
+ | 0.4378 | 3.0 | 258 | 0.3189 | 0.7263 | 0.8287 | 0.6691 | None |
58
+ | 0.2938 | 4.0 | 344 | 0.2852 | 0.7759 | 0.8435 | 0.7259 | None |
59
+ | 0.1958 | 5.0 | 430 | 0.2810 | 0.8010 | 0.8398 | 0.7726 | None |
60
+ | 0.1225 | 6.0 | 516 | 0.3017 | 0.8087 | 0.8303 | 0.7944 | None |
61
+ | 0.0715 | 7.0 | 602 | 0.2984 | 0.8206 | 0.8452 | 0.8008 | None |
62
+ | 0.0715 | 8.0 | 688 | 0.3067 | 0.8293 | 0.8420 | 0.8175 | None |
63
+ | 0.0359 | 9.0 | 774 | 0.3100 | 0.8280 | 0.8617 | 0.8002 | None |
64
+ | 0.0208 | 10.0 | 860 | 0.3137 | 0.8277 | 0.8600 | 0.8001 | None |
65
 
66
 
67
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:feca7fe02a066918a91a5710488f5c91378507a051a988b4e4b70c4abd0b793b
3
  size 437967876
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f500ab2c48bfbcd6ffc4471cdb0afe5ecca421d80fbc3bd1b87a060fa408e05
3
  size 437967876
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:747710256e7b338f73241760dbecc22c62d8385e34f76c87f6f8afc1600d607d
3
  size 5777
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9f8ea55d5fccab6b4ea950bec51e30efe1d0a414911af6c52593422e90ae309
3
  size 5777