End of training

README.md

@@ -6,9 +6,36 @@ tags:
 - generated_from_trainer
 datasets:
 - audiofolder
+metrics:
+- accuracy
+- f1
+- precision
+- recall
 model-index:
 - name: distilhubert-finetuned-cry-detector
-  results: []
+  results:
+  - task:
+      name: Audio Classification
+      type: audio-classification
+    dataset:
+      name: audiofolder
+      type: audiofolder
+      config: default
+      split: train
+      args: default
+    metrics:
+    - name: Accuracy
+      type: accuracy
+      value: 0.989010989010989
+    - name: F1
+      type: f1
+      value: 0.9890405015532383
+    - name: Precision
+      type: precision
+      value: 0.9891330367917903
+    - name: Recall
+      type: recall
+      value: 0.989010989010989
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,6 +44,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 the audiofolder dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0398
+- Accuracy: 0.9890
+- F1: 0.9890
+- Precision: 0.9891
+- Recall: 0.9890
 
 ## Model description
 
@@ -44,13 +77,15 @@ 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: 3
 
 ### Training results
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| No log        | 0.9956 | 85   | 0.1014          | 0.9641   | 0.9642 | 0.9643    | 0.9641 |
+| No log        | 0.9956 | 85   | 0.0769          | 0.9758   | 0.9760 | 0.9764    | 0.9758 |
+| No log        | 1.9912 | 170  | 0.0444          | 0.9875   | 0.9876 | 0.9876    | 0.9875 |
+| No log        | 2.9868 | 255  | 0.0398          | 0.9890   | 0.9890 | 0.9891    | 0.9890 |
 
 
 ### Framework versions

model.safetensors

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

runs/Sep02_01-09-27_ubumarcos/events.out.tfevents.1725232180.ubumarcos

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b7f9bd11ca22c7cf0512e02cfbd44cc2c93af4d753736aae8fcfab72c90bd72
+oid sha256:37edd9986d3659d5cfeb9aa8cfbe6429149a6cd74482bcfbbbae5877c8eae7df
 size 5240