Marcos12886 commited on
Commit
c0e3c8e
1 Parent(s): 26f9bb2

End of training

Browse files
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: 10
81
 
82
  ### Training results
83
 
84
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
85
  |:-------------:|:------:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
86
- | No log | 0.9032 | 7 | 1.0334 | 0.7317 | 0.6183 | 0.5354 | 0.7317 |
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": "belly_pain",
54
- "2": "burping",
55
- "3": "discomfort",
56
- "4": "hungry",
57
- "5": "pain",
58
- "6": "tired"
59
  },
60
  "initializer_range": 0.02,
61
  "intermediate_size": 3072,
62
  "label2id": {
63
  "asphyxia": "0",
64
- "belly_pain": "1",
65
- "burping": "2",
66
- "discomfort": "3",
67
- "hungry": "4",
68
- "pain": "5",
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:35d83612a6a3b6def194a62ab312c5cda812da7dc64601390eb04e937425d28f
3
- size 94768644
 
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:11af5240312a8f05e77e41d8ac5832025adf45e1241dec59641b8bf86c711aea
3
  size 5240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c67a6ea9c17797189d6d754d71b39da02a5c49105a41d05c69d958dbcf14759
3
  size 5240