End of training

README.md

@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5330
-- Accuracy: 0.8186
-- F1 Macro: 0.7907
+- Loss: 0.4057
+- Accuracy: 0.8387
+- F1 Macro: 0.8326
 
 ## Model description
 
@@ -39,9 +39,9 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 2.3824605150324688e-05
-- train_batch_size: 32
-- eval_batch_size: 32
+- learning_rate: 2e-05
+- train_batch_size: 16
+- eval_batch_size: 16
 - seed: 42
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
@@ -51,11 +51,10 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Macro |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
-| No log        | 1.0   | 461  | 0.6004          | 0.7602   | 0.6990   |
-| 0.7551        | 2.0   | 922  | 0.5176          | 0.8101   | 0.7509   |
-| 0.445         | 3.0   | 1383 | 0.4920          | 0.8199   | 0.7857   |
-| 0.3252        | 4.0   | 1844 | 0.5519          | 0.8204   | 0.7945   |
-| 0.2233        | 5.0   | 2305 | 0.5965          | 0.8079   | 0.7754   |
+| 0.7293        | 1.0   | 907  | 0.4495          | 0.8206   | 0.8107   |
+| 0.377         | 2.0   | 1814 | 0.3867          | 0.8422   | 0.8389   |
+| 0.2841        | 3.0   | 2721 | 0.4231          | 0.8383   | 0.8324   |
+| 0.1976        | 4.0   | 3628 | 0.4991          | 0.8366   | 0.8304   |
 
 
 ### Framework versions

config.json

@@ -13,17 +13,13 @@
     "1": "Normal",
     "2": "Anxiety",
     "3": "Suicidal",
-    "4": "Stress",
-    "5": "Bipolar",
-    "6": "Personality disorder"
+    "4": "Stress"
   },
   "initializer_range": 0.02,
   "label2id": {
     "Anxiety": 2,
-    "Bipolar": 5,
     "Depression": 0,
     "Normal": 1,
-    "Personality disorder": 6,
     "Stress": 4,
     "Suicidal": 3
   },

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:82745745c2198f428d333b85c74aaf66fae6c5b548519584e5dad2eb9fed69c1
-size 267847948
+oid sha256:f5e794886062b223f436e27572b8b5368627d189965a9184eb4f47e5a65b693c
+size 267841796

runs/Dec19_20-55-28_21a63788bec3/events.out.tfevents.1734641730.21a63788bec3.23.0

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

runs/Dec19_20-55-28_21a63788bec3/events.out.tfevents.1734643135.21a63788bec3.23.1

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5085e6af85619b3a62dc37cc64bbf23ccc888b9449d0aa15d0fe14ae80982d81
+oid sha256:ac088e1b74f13e2b0aaea41844c917d933472422b1c99b5475af6a9ca4f5d9cd
 size 5304