cicids2018-distilbert

README.md

@@ -3,8 +3,6 @@ license: apache-2.0
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
-metrics:
-- accuracy
 model-index:
 - name: cicids2018-distilbert
   results: []
@@ -16,9 +14,6 @@ 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.6071
-- Accuracy: 0.75
 
 ## Model description
 
@@ -43,20 +38,7 @@ 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: 7
-
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 1    | 0.6822          | 0.5      |
-| No log        | 2.0   | 2    | 0.6674          | 0.5      |
-| No log        | 3.0   | 3    | 0.6530          | 0.75     |
-| No log        | 4.0   | 4    | 0.6379          | 0.75     |
-| No log        | 5.0   | 5    | 0.6242          | 0.75     |
-| No log        | 6.0   | 6    | 0.6133          | 0.75     |
-| No log        | 7.0   | 7    | 0.6071          | 0.75     |
-
+- num_epochs: 2
 
 ### Framework versions
 

config.json

@@ -9,20 +9,19 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "NEGATIVE",
-    "1": "POSITIVE"
+    "0": "BENIGN",
+    "1": "MALICIOUS"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "NEGATIVE": 0,
-    "POSITIVE": 1
+    "BENIGN": 0,
+    "MALICIOUS": 1
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "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,

model.safetensors

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

training_args.bin

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