Hugging Face's logo

PewterZz
/
ft-bert-base-uncased-for-binary-search

Text Classification
Transformers
Safetensors
bert
Generated from Trainer
Model card Files
xet
 Community
PewterZz commited on
Commit
d7138a7
·
verified ·
1 Parent(s): 3e94ac4

End of training

Browse files
Files changed (3) hide show
  1. README.md +8 -8
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
- base_model: bert-base-uncased
3
  library_name: transformers
4
  license: apache-2.0
 
5
  tags:
6
  - generated_from_trainer
7
  model-index:
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the https://www.kaggle.com/datasets/skywardai/network-vulnerability dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.0005
20
 
21
  ## Model description
22
 
@@ -35,7 +35,7 @@ More information needed
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
- - learning_rate: 2e-05
39
  - train_batch_size: 16
40
  - eval_batch_size: 16
41
  - seed: 42
@@ -47,11 +47,11 @@ The following hyperparameters were used during training:
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
- | 0.0141 | 1.0 | 50 | 0.0061 |
51
- | 0.0016 | 2.0 | 100 | 0.0011 |
52
- | 0.001 | 3.0 | 150 | 0.0007 |
53
- | 0.0008 | 4.0 | 200 | 0.0005 |
54
- | 0.0007 | 5.0 | 250 | 0.0005 |
55
 
56
 
57
  ### Framework versions
 
1
  ---
 
2
  library_name: transformers
3
  license: apache-2.0
4
+ base_model: bert-base-uncased
5
  tags:
6
  - generated_from_trainer
7
  model-index:
 
16
 
17
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the https://www.kaggle.com/datasets/skywardai/network-vulnerability dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.0002
20
 
21
  ## Model description
22
 
 
35
  ### Training hyperparameters
36
 
37
  The following hyperparameters were used during training:
38
+ - learning_rate: 5e-05
39
  - train_batch_size: 16
40
  - eval_batch_size: 16
41
  - seed: 42
 
47
 
48
  | Training Loss | Epoch | Step | Validation Loss |
49
  |:-------------:|:-----:|:----:|:---------------:|
50
+ | 0.0012 | 1.0 | 50 | 0.0007 |
51
+ | 0.0004 | 2.0 | 100 | 0.0003 |
52
+ | 0.0003 | 3.0 | 150 | 0.0002 |
53
+ | 0.0002 | 4.0 | 200 | 0.0002 |
54
+ | 0.0002 | 5.0 | 250 | 0.0002 |
55
 
56
 
57
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e40f5031e9cd40ba0825880f84920b55eccd6ace0097026ba666341410b82c42
3
  size 437958648
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60a99da1b3e66580517353f1e692dd5b73c83188f68b9a6b0b595c04b6f98add
3
  size 437958648
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6196cecf81e92843f4dc0bcb38de8cc2ec8c80a0b9a617ea64f9d6fe05de0334
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55195b7e1ccc8a15d3ddd7645d6a35295207b04b9040ea3840347f66cf627d52
3
  size 5240