filippo571 commited on
Commit
07500c6
·
1 Parent(s): 8ded6f6

Upload TFDistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. README.md +9 -7
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,10 +14,10 @@ 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.8048
18
- - Validation Loss: 0.7373
19
- - Train Accuracy: 0.7265
20
- - Epoch: 2
21
 
22
  ## Model description
23
 
@@ -43,9 +43,11 @@ The following hyperparameters were used during training:
43
 
44
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
45
  |:----------:|:---------------:|:--------------:|:-----:|
46
- | 1.2707 | 1.0862 | 0.5812 | 0 |
47
- | 1.0411 | 0.8534 | 0.7179 | 1 |
48
- | 0.8048 | 0.7373 | 0.7265 | 2 |
 
 
49
 
50
 
51
  ### 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.5381
18
+ - Validation Loss: 0.6318
19
+ - Train Accuracy: 0.7436
20
+ - Epoch: 4
21
 
22
  ## Model description
23
 
 
43
 
44
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
45
  |:----------:|:---------------:|:--------------:|:-----:|
46
+ | 1.2425 | 1.0278 | 0.6410 | 0 |
47
+ | 0.9409 | 0.8109 | 0.6923 | 1 |
48
+ | 0.7370 | 0.6991 | 0.7436 | 2 |
49
+ | 0.6080 | 0.6474 | 0.7436 | 3 |
50
+ | 0.5381 | 0.6318 | 0.7436 | 4 |
51
 
52
 
53
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4910025126f06c42c750d2537a8e3789ef6dce548086a56a21084cec0ebb55b7
3
  size 267957952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45518e6e803ab2c9954f8821b0b18b81518290b3db18cf94db2b26d5dee3f767
3
  size 267957952