End of training

README.md

@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [huawei-noah/TinyBERT_General_4L_312D](https://huggingface.co/huawei-noah/TinyBERT_General_4L_312D) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.0050
-- Accuracy: 0.5367
+- Loss: 0.6292
+- Accuracy: 0.7664
 
 ## Model description
 
@@ -49,9 +49,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.1037        | 1.0   | 42   | 1.0993          | 0.5367   |
-| 1.0113        | 2.0   | 84   | 1.0349          | 0.5367   |
-| 0.983         | 3.0   | 126  | 1.0050          | 0.5367   |
+| 0.6875        | 1.0   | 364  | 0.7508          | 0.7226   |
+| 0.5574        | 2.0   | 728  | 0.6804          | 0.7292   |
+| 0.5481        | 3.0   | 1092 | 0.6292          | 0.7664   |
 
 
 ### Framework versions

config.json

@@ -14,15 +14,15 @@
     "0": "Advertising",
     "1": "Phishing",
     "2": "Scam",
-    "3": "Unknown"
+    "3": "Other"
   },
   "initializer_range": 0.02,
   "intermediate_size": 1200,
   "label2id": {
     "Advertising": 0,
+    "Other": 3,
     "Phishing": 1,
-    "Scam": 2,
-    "Unknown": 3
+    "Scam": 2
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d9ff30f136f81318d10cbf6aaf50fa59647b0f4d398e74711a4f3786f55866f6
+oid sha256:6353e4fdacc60c045c70d1dfcbc0c458d66b709b835184927a7bf48a1806c3b7
 size 57414312