Model save

README.md

@@ -4,9 +4,26 @@ license: bsd-3-clause
 base_model: MIT/ast-finetuned-audioset-10-10-0.4593
 tags:
 - generated_from_trainer
+datasets:
+- audiofolder
+metrics:
+- accuracy
 model-index:
 - name: ast-finetuned-audioset-10-10-0.4593-heart-sounds
-  results: []
+  results:
+  - task:
+      name: Audio Classification
+      type: audio-classification
+    dataset:
+      name: audiofolder
+      type: audiofolder
+      config: default
+      split: validation
+      args: default
+    metrics:
+    - name: Accuracy
+      type: accuracy
+      value: 0.9679878048780488
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,7 +33,10 @@ should probably proofread and complete it, then remove this comment. -->
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/vldmrl-org/HeartDiseaseDetector/runs/n9fzp3eh)
 # ast-finetuned-audioset-10-10-0.4593-heart-sounds
 
-This model is a fine-tuned version of [MIT/ast-finetuned-audioset-10-10-0.4593](https://huggingface.co/MIT/ast-finetuned-audioset-10-10-0.4593) on an unknown dataset.
+This model is a fine-tuned version of [MIT/ast-finetuned-audioset-10-10-0.4593](https://huggingface.co/MIT/ast-finetuned-audioset-10-10-0.4593) on the audiofolder dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.1499
+- Accuracy: 0.9680
 
 ## Model description
 
@@ -46,6 +66,20 @@ The following hyperparameters were used during training:
 - lr_scheduler_warmup_ratio: 0.1
 - num_epochs: 8
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 0.2135        | 1.0    | 83   | 0.1724          | 0.9390   |
+| 0.1655        | 2.0    | 166  | 0.2276          | 0.8918   |
+| 0.0959        | 3.0    | 249  | 0.1166          | 0.9527   |
+| 0.04          | 4.0    | 332  | 0.1106          | 0.9634   |
+| 0.0142        | 5.0    | 415  | 0.1317          | 0.9634   |
+| 0.0186        | 6.0    | 498  | 0.1424          | 0.9665   |
+| 0.0022        | 7.0    | 581  | 0.1433          | 0.9695   |
+| 0.0009        | 7.9119 | 656  | 0.1499          | 0.9680   |
+
+
 ### Framework versions
 
 - Transformers 4.49.0

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e79f70d6dbd4a3835641095cc144aa2185411bd0c295bda135bc93a73df62772
+oid sha256:107f370a6ebbd1154ef5323c539ca1d85f7b0dce0c4d6b372bf421bb9630a489
 size 344793116