cicids2018-distilbert

README.md

@@ -3,6 +3,9 @@ license: apache-2.0
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
+- f1
 model-index:
 - name: cicids2018-distilbert
   results: []
@@ -14,6 +17,10 @@ should probably proofread and complete it, then remove this comment. -->
 # cicids2018-distilbert
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.3903
+- Accuracy: 1.0
+- F1: 1.0
 
 ## Model description
 
@@ -40,9 +47,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 2
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
+| No log        | 1.0   | 6    | 0.4998          | 0.9167   | 0.9231 |
+| No log        | 2.0   | 12   | 0.3903          | 1.0      | 1.0    |
+
+
 ### Framework versions
 
-- Transformers 4.35.2
-- Pytorch 2.1.1+cu118
+- Transformers 4.36.2
+- Pytorch 2.1.2+cu118
 - Datasets 2.15.0
 - Tokenizers 0.15.0

config.json

@@ -22,11 +22,12 @@
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,
+  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.35.2",
+  "transformers_version": "4.36.2",
   "vocab_size": 30522
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:99fc238fcc06104edb812c139b21fb88c157d81314c5b73a8ca91cf2e3f9b46a
+oid sha256:78c58a04eef610aa9a1de563022bf0db5df9f2bc8bf4dd94f10e7ea3a132503e
 size 267832560

runs/Dec20_18-26-10_48e1c717875a/events.out.tfevents.1703096786.48e1c717875a.172.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf79f75f3f682c7627abad86e7514448cf27fcb0d713ad3a3d7ba058290003fe
+size 5438

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4126c5b5b90ade449eebd6f20307161fd202a64b00f7297a6c4fa2cb976a6379
-size 4536
+oid sha256:b8d551fc5eb223c1e88318756f349f415cf34cc27e5c5e1dfdfabdc1af5f5fdf
+size 4728