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.3259
+- Loss: 0.3128
 
 ## Model description
 
@@ -47,26 +47,15 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 0.2314        | 1.0   | 50   | 0.1855          |
-| 0.1554        | 2.0   | 100  | 0.1952          |
-| 0.2998        | 3.0   | 150  | 0.1840          |
-| 0.1898        | 4.0   | 200  | 0.1859          |
-| 0.3497        | 5.0   | 250  | 0.1881          |
-| 0.3335        | 6.0   | 300  | 0.1861          |
-| 0.221         | 7.0   | 350  | 0.1882          |
-| 0.1987        | 8.0   | 400  | 0.1883          |
-| 0.2741        | 9.0   | 450  | 0.2015          |
-| 0.268         | 10.0  | 500  | 0.2136          |
-| 0.2494        | 11.0  | 550  | 0.2406          |
-| 0.1026        | 12.0  | 600  | 0.2725          |
-| 0.1124        | 13.0  | 650  | 0.3386          |
-| 0.0798        | 14.0  | 700  | 0.3352          |
-| 0.0664        | 15.0  | 750  | 0.3259          |
+| 0.2439        | 1.0   | 50   | 0.3049          |
+| 0.3044        | 2.0   | 100  | 0.3129          |
+| 0.2049        | 3.0   | 150  | 0.3205          |
+| 0.2018        | 4.0   | 200  | 0.3128          |
 
 
 ### Framework versions
 
-- Transformers 4.44.2
-- Pytorch 2.5.0+cu121
-- Datasets 3.1.0
-- Tokenizers 0.19.1
+- Transformers 4.45.1
+- Pytorch 2.4.0
+- Datasets 3.0.1
+- Tokenizers 0.20.0

config.json

@@ -20,7 +20,7 @@
   "position_embedding_type": "absolute",
   "problem_type": "single_label_classification",
   "torch_dtype": "float32",
-  "transformers_version": "4.44.2",
+  "transformers_version": "4.45.1",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522

model.safetensors

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ba090b8cacc03bd4b5be2dda0af5e4b9ce1846af0c5886e5137e825e2aacfb1b
-size 5176
+oid sha256:16056025a0dab0a798ad60258e9b5a510861228358ff08e18ac62637a80c5ad3
+size 5240