End of training

README.md

@@ -4,6 +4,11 @@ license: apache-2.0
 base_model: ntu-spml/distilhubert
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
+- precision
+- recall
+- f1
 model-index:
 - name: distilhubert-finetuned-cry-detector
   results: []
@@ -15,6 +20,12 @@ should probably proofread and complete it, then remove this comment. -->
 # distilhubert-finetuned-cry-detector
 
 This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0878
+- Accuracy: 0.9861
+- Precision: 0.9861
+- Recall: 0.9861
+- F1: 0.9861
 
 ## Model description
 
@@ -42,13 +53,20 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.001
-- num_epochs: 1
+- num_epochs: 8
 
 ### Training results
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
 |:-------------:|:------:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
-| No log        | 0.9956 | 85   | 0.1412          | 0.9480   | 0.9478    | 0.9480 | 0.9476 |
+| No log        | 0.9956 | 85   | 0.1204          | 0.9641   | 0.9641    | 0.9641 | 0.9638 |
+| No log        | 1.9912 | 170  | 0.0847          | 0.9773   | 0.9772    | 0.9773 | 0.9773 |
+| No log        | 2.9985 | 256  | 0.1025          | 0.9766   | 0.9769    | 0.9766 | 0.9766 |
+| No log        | 3.9941 | 341  | 0.0869          | 0.9832   | 0.9835    | 0.9832 | 0.9832 |
+| No log        | 4.9898 | 426  | 0.0746          | 0.9832   | 0.9834    | 0.9832 | 0.9832 |
+| 0.0538        | 5.9971 | 512  | 0.0870          | 0.9861   | 0.9861    | 0.9861 | 0.9861 |
+| 0.0538        | 6.9927 | 597  | 0.0890          | 0.9861   | 0.9861    | 0.9861 | 0.9861 |
+| 0.0538        | 7.9649 | 680  | 0.0878          | 0.9861   | 0.9861    | 0.9861 | 0.9861 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cc34463948a1090e862c3f562483f73d9b8575c62d53f2695ead5ce878d0cb51
+oid sha256:dbc5b2463f7b10d6accd90a38663b1fe5924ce417d20bd4c4b022491ab407a53
 size 94763496

runs/Sep08_19-43-06_ubumarcos/events.out.tfevents.1725817387.ubumarcos

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

runs/Sep08_19-57-45_ubumarcos/events.out.tfevents.1725818266.ubumarcos

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e03a8d2517098878f32d3d0862d22bbbeeb4ecf145573639f810ef31e34b5336
+oid sha256:cda0bf828ccc04cd6d3fb7933f60f34ca3c8ba14b14579f9e878d688bc739b4a
 size 5240