End of training

README.md

@@ -4,8 +4,6 @@ license: apache-2.0
 base_model: ntu-spml/distilhubert
 tags:
 - generated_from_trainer
-datasets:
-- audiofolder
 model-index:
 - name: distilhubert-finetuned-mixed-data
   results: []
@@ -16,7 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilhubert-finetuned-mixed-data
 
-This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
+This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on an unknown dataset.
 
 ## Model description
 
@@ -48,14 +46,13 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
-|:-------------:|:------:|:----:|:---------------:|:--------:|
-| No log        | 0.9664 | 18   | 0.9747          | 0.7282   |
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
+|:-------------:|:------:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
+| No log        | 0.9897 | 24   | 1.1111          | 0.5401   | 0.5537    | 0.5401 | 0.4192 |
 
 
 ### Framework versions
 
 - Transformers 4.44.2
-- Pytorch 2.4.0+cu121
-- Datasets 2.21.0
+- Pytorch 2.4.1+cu121
 - Tokenizers 0.19.1

config.json

@@ -50,18 +50,18 @@
   "hidden_dropout": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "1s_asphyxia",
-    "1": "1s_hunger",
-    "2": "1s_normal",
-    "3": "1s_pain"
+    "0": "1s_normal",
+    "1": "1s_pain",
+    "2": "1s_hunger",
+    "3": "1s_asphyxia"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "1s_asphyxia": "0",
-    "1s_hunger": "1",
-    "1s_normal": "2",
-    "1s_pain": "3"
+    "1s_asphyxia": 3,
+    "1s_hunger": 2,
+    "1s_normal": 0,
+    "1s_pain": 1
   },
   "layer_norm_eps": 1e-05,
   "layerdrop": 0.0,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4843ef1011fb35e92cba9dc2f9cc3fd67496f3f6e43139cdcad8ac5517dcef8c
+oid sha256:3c4a38a8731592e9ea57f1c93bd70721303a74c27c23536f1c5becd511e1b447
 size 94765560

runs/Sep08_17-35-11_ubumarcos/events.out.tfevents.1725809712.ubumarcos

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

runs/Sep08_17-35-57_ubumarcos/events.out.tfevents.1725809757.ubumarcos

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

runs/Sep08_17-36-57_ubumarcos/events.out.tfevents.1725809817.ubumarcos

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37227eef9f32c4f5587a8c395af1f4480ae1f56398047e7e00408639044cb25c
+oid sha256:997e4fa07d4013eed6ee03d5e8203f70e0b8c95816857b594665b938216b48a7
 size 5240