Marcos12886
commited on
Commit
•
c0e3c8e
1
Parent(s):
26f9bb2
End of training
Browse files- README.md +3 -44
- config.json +10 -12
- model.safetensors +2 -2
- runs/Aug29_16-18-04_DesMar/events.out.tfevents.1724941094.DesMar +3 -0
- training_args.bin +1 -1
README.md
CHANGED
@@ -6,36 +6,9 @@ tags:
|
|
6 |
- generated_from_trainer
|
7 |
datasets:
|
8 |
- audiofolder
|
9 |
-
metrics:
|
10 |
-
- accuracy
|
11 |
-
- f1
|
12 |
-
- precision
|
13 |
-
- recall
|
14 |
model-index:
|
15 |
- name: distilhubert-finetuned-donateacry
|
16 |
-
results:
|
17 |
-
- task:
|
18 |
-
name: Audio Classification
|
19 |
-
type: audio-classification
|
20 |
-
dataset:
|
21 |
-
name: audiofolder
|
22 |
-
type: audiofolder
|
23 |
-
config: default
|
24 |
-
split: train
|
25 |
-
args: default
|
26 |
-
metrics:
|
27 |
-
- name: Accuracy
|
28 |
-
type: accuracy
|
29 |
-
value: 0.8130081300813008
|
30 |
-
- name: F1
|
31 |
-
type: f1
|
32 |
-
value: 0.7606844060819746
|
33 |
-
- name: Precision
|
34 |
-
type: precision
|
35 |
-
value: 0.7167376435669118
|
36 |
-
- name: Recall
|
37 |
-
type: recall
|
38 |
-
value: 0.8130081300813008
|
39 |
---
|
40 |
|
41 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
@@ -44,12 +17,6 @@ should probably proofread and complete it, then remove this comment. -->
|
|
44 |
# distilhubert-finetuned-donateacry
|
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.7024
|
49 |
-
- Accuracy: 0.8130
|
50 |
-
- F1: 0.7607
|
51 |
-
- Precision: 0.7167
|
52 |
-
- Recall: 0.8130
|
53 |
|
54 |
## Model description
|
55 |
|
@@ -77,21 +44,13 @@ 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.03
|
80 |
-
- num_epochs:
|
81 |
|
82 |
### Training results
|
83 |
|
84 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
|
85 |
|:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
|
86 |
-
| No log | 0.9032 | 7 | 1.
|
87 |
-
| No log | 1.9355 | 15 | 0.9193 | 0.7642 | 0.6866 | 0.6238 | 0.7642 |
|
88 |
-
| No log | 2.9677 | 23 | 0.7766 | 0.8049 | 0.7460 | 0.7005 | 0.8049 |
|
89 |
-
| No log | 4.0 | 31 | 0.8394 | 0.7724 | 0.7275 | 0.6889 | 0.7724 |
|
90 |
-
| No log | 4.9032 | 38 | 0.7391 | 0.7805 | 0.7351 | 0.6962 | 0.7805 |
|
91 |
-
| No log | 5.9355 | 46 | 0.7578 | 0.8130 | 0.7607 | 0.7167 | 0.8130 |
|
92 |
-
| No log | 6.9677 | 54 | 0.6822 | 0.8049 | 0.7558 | 0.7147 | 0.8049 |
|
93 |
-
| No log | 8.0 | 62 | 0.6980 | 0.8049 | 0.7543 | 0.7119 | 0.8049 |
|
94 |
-
| No log | 8.9032 | 69 | 0.7024 | 0.8130 | 0.7607 | 0.7167 | 0.8130 |
|
95 |
|
96 |
|
97 |
### Framework versions
|
|
|
6 |
- generated_from_trainer
|
7 |
datasets:
|
8 |
- audiofolder
|
|
|
|
|
|
|
|
|
|
|
9 |
model-index:
|
10 |
- name: distilhubert-finetuned-donateacry
|
11 |
+
results: []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
---
|
13 |
|
14 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
17 |
# distilhubert-finetuned-donateacry
|
18 |
|
19 |
This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
## Model description
|
22 |
|
|
|
44 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
45 |
- lr_scheduler_type: cosine
|
46 |
- lr_scheduler_warmup_ratio: 0.03
|
47 |
+
- num_epochs: 1
|
48 |
|
49 |
### Training results
|
50 |
|
51 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
|
52 |
|:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
|
53 |
+
| No log | 0.9032 | 7 | 1.0005 | 0.7317 | 0.6183 | 0.5354 | 0.7317 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
54 |
|
55 |
|
56 |
### Framework versions
|
config.json
CHANGED
@@ -50,23 +50,21 @@
|
|
50 |
"hidden_size": 768,
|
51 |
"id2label": {
|
52 |
"0": "asphyxia",
|
53 |
-
"1": "
|
54 |
-
"2": "
|
55 |
-
"3": "
|
56 |
-
"4": "
|
57 |
-
"5": "
|
58 |
-
"6": "tired"
|
59 |
},
|
60 |
"initializer_range": 0.02,
|
61 |
"intermediate_size": 3072,
|
62 |
"label2id": {
|
63 |
"asphyxia": "0",
|
64 |
-
"
|
65 |
-
"
|
66 |
-
"
|
67 |
-
"
|
68 |
-
"
|
69 |
-
"tired": "6"
|
70 |
},
|
71 |
"layer_norm_eps": 1e-05,
|
72 |
"layerdrop": 0.0,
|
|
|
50 |
"hidden_size": 768,
|
51 |
"id2label": {
|
52 |
"0": "asphyxia",
|
53 |
+
"1": "burping",
|
54 |
+
"2": "discomfort",
|
55 |
+
"3": "hungry",
|
56 |
+
"4": "pain",
|
57 |
+
"5": "tired"
|
|
|
58 |
},
|
59 |
"initializer_range": 0.02,
|
60 |
"intermediate_size": 3072,
|
61 |
"label2id": {
|
62 |
"asphyxia": "0",
|
63 |
+
"burping": "1",
|
64 |
+
"discomfort": "2",
|
65 |
+
"hungry": "3",
|
66 |
+
"pain": "4",
|
67 |
+
"tired": "5"
|
|
|
68 |
},
|
69 |
"layer_norm_eps": 1e-05,
|
70 |
"layerdrop": 0.0,
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ee89457dfabcb5266a0d9ddd072149adddb23b662b81849ffec13e9805e3969d
|
3 |
+
size 94767616
|
runs/Aug29_16-18-04_DesMar/events.out.tfevents.1724941094.DesMar
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0b13ae1a2f422bdd96d5dbb3f41e9b701af935546936d2d1d491455c43a614e0
|
3 |
+
size 6765
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 5240
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7c67a6ea9c17797189d6d754d71b39da02a5c49105a41d05c69d958dbcf14759
|
3 |
size 5240
|