Hugging Face's logo

alexpaunoiu
/
bert_finetune_prompt_classification

Text Classification
Transformers
PyTorch
bert
Generated from Trainer
Inference Endpoints
Model card Files Community
alexpaunoiu commited on
Commit
109d63f
·
1 Parent(s): 5aca58d

Model save

Browse files
Files changed (4) hide show
  1. README.md +13 -13
  2. config.json +1 -1
  3. pytorch_model.bin +2 -2
  4. training_args.bin +1 -1
README.md CHANGED
@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
- - Loss: 0.1441
19
 
20
  ## Model description
21
 
@@ -49,18 +49,18 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:-----:|:----:|:---------------:|
52
- | 0.8543 | 0.98 | 32 | 0.8580 |
53
- | 0.5207 | 2.0 | 65 | 0.5211 |
54
- | 0.3075 | 2.98 | 97 | 0.3217 |
55
- | 0.1846 | 4.0 | 130 | 0.2415 |
56
- | 0.2073 | 4.98 | 162 | 0.2145 |
57
- | 0.1563 | 6.0 | 195 | 0.1965 |
58
- | 0.0591 | 6.98 | 227 | 0.1602 |
59
- | 0.0791 | 8.0 | 260 | 0.1575 |
60
- | 0.0528 | 8.98 | 292 | 0.1717 |
61
- | 0.0977 | 10.0 | 325 | 0.1458 |
62
- | 0.0708 | 10.98 | 357 | 0.1440 |
63
- | 0.1147 | 11.82 | 384 | 0.1441 |
64
 
65
 
66
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.1537
19
 
20
  ## Model description
21
 
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss |
51
  |:-------------:|:-----:|:----:|:---------------:|
52
+ | 0.945 | 1.0 | 33 | 0.9316 |
53
+ | 0.5897 | 1.99 | 66 | 0.3983 |
54
+ | 0.2354 | 2.99 | 99 | 0.2213 |
55
+ | 0.237 | 3.98 | 132 | 0.1955 |
56
+ | 0.2126 | 4.98 | 165 | 0.1485 |
57
+ | 0.124 | 5.98 | 198 | 0.1412 |
58
+ | 0.1857 | 6.97 | 231 | 0.1540 |
59
+ | 0.108 | 8.0 | 265 | 0.1602 |
60
+ | 0.0801 | 9.0 | 298 | 0.1561 |
61
+ | 0.0421 | 9.99 | 331 | 0.1430 |
62
+ | 0.015 | 10.99 | 364 | 0.1480 |
63
+ | 0.0113 | 11.95 | 396 | 0.1537 |
64
 
65
 
66
  ### Framework versions
config.json CHANGED
@@ -33,5 +33,5 @@
33
  "transformers_version": "4.34.1",
34
  "type_vocab_size": 2,
35
  "use_cache": true,
36
- "vocab_size": 36954
37
  }
 
33
  "transformers_version": "4.34.1",
34
  "type_vocab_size": 2,
35
  "use_cache": true,
36
+ "vocab_size": 36969
37
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4dfbbd8ec7a29b7722d0ff78b1a7a8f2638e9342fb0cb4ddcc56869719538037
3
- size 457765681
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:575a83e93bb7d4f05134b65f27c12693945866745366fc906e76abff7ec3b2c2
3
+ size 457811761
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:67dba09312b2430c6dd5761457b7c94ed15510bea6b3fe83f7c9a16d88591066
3
  size 4155
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d7f7c9914ebd8e38842a50b4cfd2e9cb89cc3cf9249ab6d9275fbe408f38bc1
3
  size 4155