End of training

README.md

@@ -1,7 +1,7 @@
 ---
-base_model: bert-base-uncased
 library_name: transformers
 license: apache-2.0
+base_model: bert-base-uncased
 tags:
 - generated_from_trainer
 model-index:
@@ -16,7 +16,7 @@ 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 https://www.kaggle.com/datasets/skywardai/network-vulnerability dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2320
+- Loss: 0.2520
 
 ## Model description
 
@@ -47,9 +47,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 0.1811        | 1.0   | 63   | 0.2328          |
-| 0.2025        | 2.0   | 126  | 0.2350          |
-| 0.1764        | 3.0   | 189  | 0.2320          |
+| 0.2417        | 1.0   | 63   | 0.2539          |
+| 0.1808        | 2.0   | 126  | 0.2488          |
+| 0.1045        | 3.0   | 189  | 0.2520          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:99e8ee84bfc6f6492ef8c70e3eaf1a15f0d84663d986d9b4f01ae098802974b1
+oid sha256:e84089372be791912ba781d7b0f22a61d52260d1e9a21d40c27140579bcbf8db
 size 437958648