Hugging Face's logo

leyuxzhang
/
wav2vec2-base-finetuned-iemocap

Audio Classification
Transformers
TensorBoard
Safetensors
wav2vec2
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
leyuxzhang commited on
Commit
6f9c58b
•
1 Parent(s): fc0de73

IEMOCAP Speech Emotion Classification with Wav2Vec2 Example

Browse files
Files changed (2) hide show
  1. README.md +14 -8
  2. model.safetensors +1 -1
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.7788
21
- - Accuracy: 0.7269
22
 
23
  ## Model description
24
 
@@ -43,17 +43,23 @@ The following hyperparameters were used during training:
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
- - num_epochs: 5
 
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
- | 0.8185 | 1.0 | 139 | 0.8479 | 0.6673 |
53
- | 0.6511 | 2.0 | 278 | 0.7530 | 0.7306 |
54
- | 0.545 | 3.0 | 417 | 0.7660 | 0.6980 |
55
- | 0.444 | 4.0 | 556 | 0.7803 | 0.7288 |
56
- | 0.3698 | 5.0 | 695 | 0.7788 | 0.7269 |
 
 
 
 
 
57
 
58
 
59
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.9498
21
+ - Accuracy: 0.6799
22
 
23
  ## Model description
24
 
 
43
  - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
+ - lr_scheduler_warmup_steps: 100
47
+ - num_epochs: 10
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 1.2595 | 1.0 | 139 | 1.2167 | 0.4177 |
54
+ | 1.0205 | 2.0 | 278 | 0.9921 | 0.5823 |
55
+ | 0.8923 | 3.0 | 417 | 0.9213 | 0.6040 |
56
+ | 0.7682 | 4.0 | 556 | 1.0543 | 0.5895 |
57
+ | 0.6554 | 5.0 | 695 | 0.9086 | 0.6420 |
58
+ | 0.5389 | 6.0 | 834 | 0.8520 | 0.6817 |
59
+ | 0.4351 | 7.0 | 973 | 0.9517 | 0.6745 |
60
+ | 0.3707 | 8.0 | 1112 | 0.8934 | 0.6908 |
61
+ | 0.3057 | 9.0 | 1251 | 0.9342 | 0.6781 |
62
+ | 0.2566 | 10.0 | 1390 | 0.9498 | 0.6799 |
63
 
64
 
65
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0f4f36b26bfb2239dc9891a7120c289c2de762e96cf5da95f8bb51583dcd689b
3
  size 378304424
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:abd5221b7b7d3159fe13a10ac4b715cbca170b3fe4ff9c626c718f50a927217f
3
  size 378304424