Hemg
/

dangerous-heartbeat-detection

Audio Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Hemg commited on Mar 31

Commit

607fd5e

•

1 Parent(s): d34a7de

Model save

Files changed (2) hide show

README.md +6 -6
model.safetensors +1 -1

README.md CHANGED Viewed

@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the audiofolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1647
 - Accuracy: 1.0
 ## Model description
@@ -52,7 +52,7 @@ More information needed
 ### Training hyperparameters
 The following hyperparameters were used during training:
-- learning_rate: 0.0005
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
@@ -67,10 +67,10 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.7573        | 0.89  | 2    | 1.2997          | 1.0      |
-| 1.0518        | 1.78  | 4    | 0.5840          | 1.0      |
-| 0.4723        | 2.67  | 6    | 0.2595          | 1.0      |
-| 0.1425        | 3.56  | 8    | 0.1647          | 1.0      |
 ### Framework versions

 This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the audiofolder dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.2805
 - Accuracy: 1.0
 ## Model description
 ### Training hyperparameters
 The following hyperparameters were used during training:
+- learning_rate: 0.0003
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 1.8144        | 0.89  | 2    | 1.4061          | 1.0      |
+| 1.2223        | 1.78  | 4    | 0.6946          | 1.0      |
+| 0.5897        | 2.67  | 6    | 0.3691          | 1.0      |
+| 0.2163        | 3.56  | 8    | 0.2805          | 1.0      |
 ### Framework versions

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a6db263350e5dcf65a765455676b92d61e30e6dd971eeb6ebcef43ee398f023d
 size 378305452

 version https://git-lfs.github.com/spec/v1
+oid sha256:03dfeeb05be156bb8cf7aeac64bb53fc370719eeb5c7642048395408ce93f02d
 size 378305452