Hugging Face's logo

doyoungkim
/
bert-base-uncased-finetuned-sst2

Text Classification
Transformers
PyTorch
TensorBoard
bert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
3
doyoungkim commited on
Commit
e3a1145
•
1 Parent(s): 1a302ea

fine tuned, probability as logits

Browse files
Files changed (5) hide show
  1. README.md +9 -9
  2. config.json +1 -1
  3. pytorch_model.bin +1 -1
  4. tokenizer.json +0 -0
  5. training_args.bin +1 -1
README.md CHANGED
@@ -16,7 +16,7 @@ model_index:
16
  metric:
17
  name: Accuracy
18
  type: accuracy
19
- value: 0.930045871559633
20
  ---
21
 
22
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -26,8 +26,8 @@ should probably proofread and complete it, then remove this comment. -->
26
 
27
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the glue dataset.
28
  It achieves the following results on the evaluation set:
29
- - Loss: 0.2485
30
- - Accuracy: 0.9300
31
 
32
  ## Model description
33
 
@@ -58,16 +58,16 @@ The following hyperparameters were used during training:
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
60
  |:-------------:|:-----:|:-----:|:---------------:|:--------:|
61
- | 0.1662 | 1.0 | 2105 | 0.2485 | 0.9300 |
62
- | 0.1102 | 2.0 | 4210 | 0.2777 | 0.9266 |
63
- | 0.0835 | 3.0 | 6315 | 0.3368 | 0.9232 |
64
- | 0.0529 | 4.0 | 8420 | 0.3310 | 0.9255 |
65
- | 0.035 | 5.0 | 10525 | 0.3855 | 0.9278 |
66
 
67
 
68
  ### Framework versions
69
 
70
- - Transformers 4.9.1
71
  - Pytorch 1.8.1
72
  - Datasets 1.11.0
73
  - Tokenizers 0.10.1
 
16
  metric:
17
  name: Accuracy
18
  type: accuracy
19
+ value: 0.926605504587156
20
  ---
21
 
22
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
26
 
27
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the glue dataset.
28
  It achieves the following results on the evaluation set:
29
+ - Loss: 0.2716
30
+ - Accuracy: 0.9266
31
 
32
  ## Model description
33
 
 
58
 
59
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
60
  |:-------------:|:-----:|:-----:|:---------------:|:--------:|
61
+ | 0.1666 | 1.0 | 2105 | 0.2403 | 0.9232 |
62
+ | 0.1122 | 2.0 | 4210 | 0.2716 | 0.9266 |
63
+ | 0.0852 | 3.0 | 6315 | 0.3150 | 0.9232 |
64
+ | 0.056 | 4.0 | 8420 | 0.3209 | 0.9163 |
65
+ | 0.0344 | 5.0 | 10525 | 0.3740 | 0.9243 |
66
 
67
 
68
  ### Framework versions
69
 
70
+ - Transformers 4.9.2
71
  - Pytorch 1.8.1
72
  - Datasets 1.11.0
73
  - Tokenizers 0.10.1
config.json CHANGED
@@ -20,7 +20,7 @@
20
  "pretrained_model_name_or_path": "bert-base-uncased",
21
  "problem_type": "single_label_classification",
22
  "torch_dtype": "float32",
23
- "transformers_version": "4.9.1",
24
  "type_vocab_size": 2,
25
  "use_cache": true,
26
  "vocab_size": 30522
 
20
  "pretrained_model_name_or_path": "bert-base-uncased",
21
  "problem_type": "single_label_classification",
22
  "torch_dtype": "float32",
23
+ "transformers_version": "4.9.2",
24
  "type_vocab_size": 2,
25
  "use_cache": true,
26
  "vocab_size": 30522
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2fb1ed9f505fd6850ca4e7dc62090e574b21bddc86f90dc84244483c55ff00e6
3
  size 438024457
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08529ab64918137e60028b1db6d71a974e89768657a53720df6dba21499433e6
3
  size 438024457
tokenizer.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f4bd401f7f9e1dec32cdf3725c9316fc6e4df3e3cde119445a4a25eb2c1b8e33
3
  size 2671
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdb5057e819519716b9f578fcdaebe281cb15630cdb063a47cf41f69fb884138
3
  size 2671