Steven Liu commited on
Commit
3bee18f
1 Parent(s): 0b80543

End of training

Browse files
Files changed (2) hide show
  1. README.md +9 -7
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,13 +14,13 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 0.3233
18
- - Validation Loss: 0.3099
19
- - Train Precision: 0.4155
20
- - Train Recall: 0.2117
21
- - Train F1: 0.2805
22
- - Train Accuracy: 0.9333
23
- - Epoch: 0
24
 
25
  ## Model description
26
 
@@ -47,6 +47,8 @@ The following hyperparameters were used during training:
47
  | Train Loss | Validation Loss | Train Precision | Train Recall | Train F1 | Train Accuracy | Epoch |
48
  |:----------:|:---------------:|:---------------:|:------------:|:--------:|:--------------:|:-----:|
49
  | 0.3233 | 0.3099 | 0.4155 | 0.2117 | 0.2805 | 0.9333 | 0 |
 
 
50
 
51
 
52
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 0.1210
18
+ - Validation Loss: 0.2698
19
+ - Train Precision: 0.5099
20
+ - Train Recall: 0.3995
21
+ - Train F1: 0.4480
22
+ - Train Accuracy: 0.9444
23
+ - Epoch: 2
24
 
25
  ## Model description
26
 
 
47
  | Train Loss | Validation Loss | Train Precision | Train Recall | Train F1 | Train Accuracy | Epoch |
48
  |:----------:|:---------------:|:---------------:|:------------:|:--------:|:--------------:|:-----:|
49
  | 0.3233 | 0.3099 | 0.4155 | 0.2117 | 0.2805 | 0.9333 | 0 |
50
+ | 0.1600 | 0.2743 | 0.5111 | 0.3589 | 0.4216 | 0.9416 | 1 |
51
+ | 0.1210 | 0.2698 | 0.5099 | 0.3995 | 0.4480 | 0.9444 | 2 |
52
 
53
 
54
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d09d6aaaf3f1cdae11cc3bec9be4c2b37de6bc816589862737768c7a9b3c362
3
  size 265618792
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:226168bc6bdf82fb0c38a9f42cdd477d9403305f891bb0f5b50c9affe7f940ee
3
  size 265618792