nuwrong commited on
Commit
2b7b2e1
1 Parent(s): ef4e5ba

Training completed!

Browse files
Files changed (2) hide show
  1. README.md +5 -5
  2. pytorch_model.bin +1 -1
README.md CHANGED
@@ -25,7 +25,7 @@ model-index:
25
  value: 0.901
26
  - name: F1
27
  type: f1
28
- value: 0.8966795034198314
29
  ---
30
 
31
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -35,9 +35,9 @@ should probably proofread and complete it, then remove this comment. -->
35
 
36
  This model is a fine-tuned version of [nuwrong/distilbert-base-uncased-finetuned-emotion](https://huggingface.co/nuwrong/distilbert-base-uncased-finetuned-emotion) on the emotion dataset.
37
  It achieves the following results on the evaluation set:
38
- - Loss: 0.3550
39
  - Accuracy: 0.901
40
- - F1: 0.8967
41
 
42
  ## Model description
43
 
@@ -68,8 +68,8 @@ The following hyperparameters were used during training:
68
 
69
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
70
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
71
- | No log | 1.0 | 125 | 0.5711 | 0.812 | 0.7784 |
72
- | 0.7662 | 2.0 | 250 | 0.3550 | 0.901 | 0.8967 |
73
 
74
 
75
  ### Framework versions
 
25
  value: 0.901
26
  - name: F1
27
  type: f1
28
+ value: 0.89754070016765
29
  ---
30
 
31
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
35
 
36
  This model is a fine-tuned version of [nuwrong/distilbert-base-uncased-finetuned-emotion](https://huggingface.co/nuwrong/distilbert-base-uncased-finetuned-emotion) on the emotion dataset.
37
  It achieves the following results on the evaluation set:
38
+ - Loss: 0.3518
39
  - Accuracy: 0.901
40
+ - F1: 0.8975
41
 
42
  ## Model description
43
 
 
68
 
69
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
70
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
71
+ | No log | 1.0 | 125 | 0.5649 | 0.821 | 0.7944 |
72
+ | 0.7652 | 2.0 | 250 | 0.3518 | 0.901 | 0.8975 |
73
 
74
 
75
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:523f93154c29c33e916386625138392e71ee927f799e3634b99f1d7bdcb7da35
3
  size 267867821
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59d916d8f10ab990d4dd16c7f2a7b7f0885621801c8a9b75fd7e61d6f2393597
3
  size 267867821