Hemg
/

heartbeat-detection

Audio Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Hemg commited on Apr 4

Commit

40518b6

•

1 Parent(s): 4a31bdd

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 [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0188
 - Accuracy: 1.0
 ## Model description
@@ -58,17 +58,17 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_ratio: 0.01
 - num_epochs: 8
 ### Training results
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.0887        | 1.0   | 9    | 0.4748          | 1.0      |
-| 0.3246        | 2.0   | 18   | 0.1406          | 1.0      |
-| 0.1064        | 3.0   | 27   | 0.0461          | 1.0      |
-| 0.0386        | 4.0   | 36   | 0.0188          | 1.0      |
 ### Framework versions

 This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.0223
 - Accuracy: 1.0
 ## Model description
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
+- lr_scheduler_warmup_ratio: 0.1
 - num_epochs: 8
 ### Training results
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 1.406         | 1.0   | 9    | 0.8395          | 1.0      |
+| 0.5234        | 2.0   | 18   | 0.2155          | 1.0      |
+| 0.1532        | 3.0   | 27   | 0.0627          | 1.0      |
+| 0.05          | 4.0   | 36   | 0.0223          | 1.0      |
 ### Framework versions

model.safetensors CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:1835f04a23145e5580538a13a094e45ae834304e8dfe86ffe3f1bc3c656bd761
 size 94766588