Hugging Face's logo

Marcos12886
/
distilhubert-finetuned-cry-detector

Audio Classification
Transformers
TensorBoard
Safetensors
hubert
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
Marcos12886 commited on
Commit
cc30eae
•
1 Parent(s): 4625a94

End of training

Browse files
Files changed (4) hide show
  1. README.md +17 -13
  2. model.safetensors +1 -1
  3. runs/Sep02_11-35-27_ubumarcos/events.out.tfevents.1725269735.ubumarcos +3 -0
  4. training_args.bin +1 -1
README.md CHANGED
@@ -26,16 +26,16 @@ model-index:
26
  metrics:
27
  - name: Accuracy
28
  type: accuracy
29
- value: 0.989010989010989
30
  - name: F1
31
  type: f1
32
- value: 0.9890405015532383
33
  - name: Precision
34
  type: precision
35
- value: 0.9891330367917903
36
  - name: Recall
37
  type: recall
38
- value: 0.989010989010989
39
  ---
40
 
41
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -45,11 +45,11 @@ should probably proofread and complete it, then remove this comment. -->
45
 
46
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
47
  It achieves the following results on the evaluation set:
48
- - Loss: 0.0398
49
- - Accuracy: 0.9890
50
- - F1: 0.9890
51
- - Precision: 0.9891
52
- - Recall: 0.9890
53
 
54
  ## Model description
55
 
@@ -77,15 +77,19 @@ The following hyperparameters were used during training:
77
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
78
  - lr_scheduler_type: cosine
79
  - lr_scheduler_warmup_ratio: 0.001
80
- - num_epochs: 3
81
 
82
  ### Training results
83
 
84
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
85
  |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
86
- | No log | 0.9956 | 85 | 0.0769 | 0.9758 | 0.9760 | 0.9764 | 0.9758 |
87
- | No log | 1.9912 | 170 | 0.0444 | 0.9875 | 0.9876 | 0.9876 | 0.9875 |
88
- | No log | 2.9868 | 255 | 0.0398 | 0.9890 | 0.9890 | 0.9891 | 0.9890 |
 
 
 
 
89
 
90
 
91
  ### Framework versions
 
26
  metrics:
27
  - name: Accuracy
28
  type: accuracy
29
+ value: 0.991941391941392
30
  - name: F1
31
  type: f1
32
+ value: 0.9919569277165429
33
  - name: Precision
34
  type: precision
35
+ value: 0.9920048531706146
36
  - name: Recall
37
  type: recall
38
+ value: 0.991941391941392
39
  ---
40
 
41
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
45
 
46
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
47
  It achieves the following results on the evaluation set:
48
+ - Loss: 0.0408
49
+ - Accuracy: 0.9919
50
+ - F1: 0.9920
51
+ - Precision: 0.9920
52
+ - Recall: 0.9919
53
 
54
  ## Model description
55
 
 
77
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
78
  - lr_scheduler_type: cosine
79
  - lr_scheduler_warmup_ratio: 0.001
80
+ - num_epochs: 7
81
 
82
  ### Training results
83
 
84
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
85
  |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
86
+ | No log | 0.9956 | 85 | 0.0736 | 0.9788 | 0.9788 | 0.9790 | 0.9788 |
87
+ | No log | 1.9912 | 170 | 0.0680 | 0.9758 | 0.9760 | 0.9770 | 0.9758 |
88
+ | No log | 2.9985 | 256 | 0.0447 | 0.9875 | 0.9876 | 0.9876 | 0.9875 |
89
+ | No log | 3.9941 | 341 | 0.0452 | 0.9905 | 0.9905 | 0.9905 | 0.9905 |
90
+ | No log | 4.9898 | 426 | 0.0439 | 0.9919 | 0.9920 | 0.9920 | 0.9919 |
91
+ | 0.053 | 5.9971 | 512 | 0.0401 | 0.9919 | 0.9920 | 0.9920 | 0.9919 |
92
+ | 0.053 | 6.9693 | 595 | 0.0408 | 0.9919 | 0.9920 | 0.9920 | 0.9919 |
93
 
94
 
95
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9f4d1bcece15f285a46eb855f203bd43924522bb36fe48d2d6c26990580d046d
3
  size 94763496
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94760f7df9dc2c6bad702593ec0871720fa6ae352e88f79c71bcfccc8fab1a09
3
  size 94763496
runs/Sep02_11-35-27_ubumarcos/events.out.tfevents.1725269735.ubumarcos ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eea5820ea966730d8e65a4c150b9a7ba53beaed1521273e658e1d9f112bad09a
3
+ size 9669
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:37edd9986d3659d5cfeb9aa8cfbe6429149a6cd74482bcfbbbae5877c8eae7df
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:caa3df7f2810faa9c35a46d178b12337edb0b134aa27dc1689b302d9bcc9d3d0
3
  size 5240