rjac commited on
Commit
e404c3a
·
1 Parent(s): 225f90d

Upload TFDistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. README.md +6 -6
  2. tf_model.h5 +2 -2
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.5063
18
- - Train Accuracy: 0.8805
19
- - Validation Loss: 0.5579
20
- - Validation Accuracy: 0.8622
21
  - Epoch: 0
22
 
23
  ## Model description
@@ -37,14 +37,14 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 636, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False}
41
  - training_precision: float32
42
 
43
  ### Training results
44
 
45
  | Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
46
  |:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
47
- | 0.5063 | 0.8805 | 0.5579 | 0.8622 | 0 |
48
 
49
 
50
  ### 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: 1.8575
18
+ - Train Accuracy: 0.5786
19
+ - Validation Loss: 0.8849
20
+ - Validation Accuracy: 0.8171
21
  - Epoch: 0
22
 
23
  ## Model description
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 2120, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False}
41
  - training_precision: float32
42
 
43
  ### Training results
44
 
45
  | Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
46
  |:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
47
+ | 1.8575 | 0.5786 | 0.8849 | 0.8171 | 0 |
48
 
49
 
50
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8048fcc98121be82c51d648df51c6a92b08694d92cde78dae6ae84bf87c77c6c
3
- size 268007104
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa6ab228a0ff497a060043363b7955273bebc4ab4bdd7f3a15480c4eea859c48
3
+ size 268010440