End of training

README.md

@@ -1,11 +1,11 @@
 ---
-base_model: bert-base-uncased
 library_name: transformers
 license: apache-2.0
-metrics:
-- accuracy
+base_model: bert-base-uncased
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: Balanced-KFold-ft-bert-base-uncased-for-binary-search
   results: []
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the network_vulnerability_dataset.csv dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5028
-- Accuracy: 0.7256
+- Loss: 0.4549
+- Accuracy: 0.7989
 
 ## Model description
 
@@ -44,16 +44,16 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 8
+- num_epochs: 10
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.4316        | 1.0   | 87   | 0.5028          | 0.7256   |
-| 0.5173        | 2.0   | 174  | 0.5019          | 0.7256   |
-| 0.4516        | 3.0   | 261  | 0.5064          | 0.7256   |
-| 0.4646        | 4.0   | 348  | 0.5050          | 0.7256   |
+| 0.4663        | 1.0   | 109  | 0.4549          | 0.7989   |
+| 0.4814        | 2.0   | 218  | 0.4474          | 0.7989   |
+| 0.4625        | 3.0   | 327  | 0.4467          | 0.7989   |
+| 0.523         | 4.0   | 436  | 0.4532          | 0.7989   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8d225cb93db9bbbba23d9fecda847e349741f6f243921cf982bbd35d859ba406
+oid sha256:172ae410600f3bc4aecd7e1974cd1be1ff0e2ab6fea64361edc467d6153dfb3c
 size 437958648

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9a4812ece9848aacc8e0d93b957ccdbd48e7c8ac2bd4fa9225ce310afa0a9e93
+oid sha256:4cea4970d271b42018dcaaa7068cfbcd28dedf14d957816e350aa29d3db121fb
 size 5240