DavidFM43
/

distilhubert-finetuned-gtzan

Audio Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

DavidFM43 commited on Jul 18, 2023

Commit

e627aae

•

1 Parent(s): c31e4fd

update model card README.md

Files changed (1) hide show

README.md +6 -6

README.md CHANGED Viewed

@@ -18,7 +18,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 GTZAN dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6174
 - Accuracy: 0.84
 ## Model description
@@ -53,11 +53,11 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 1.2016        | 1.0   | 112  | 1.2866          | 0.57     |
-| 0.7752        | 2.0   | 225  | 0.9609          | 0.74     |
-| 0.7032        | 3.0   | 337  | 0.9225          | 0.7      |
-| 0.4642        | 4.0   | 450  | 0.8619          | 0.79     |
-| 0.2253        | 4.98  | 560  | 0.6174          | 0.84     |
 ### Framework versions

 This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.6152
 - Accuracy: 0.84
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 1.2014        | 1.0   | 112  | 1.2862          | 0.57     |
+| 0.7805        | 2.0   | 225  | 0.9564          | 0.72     |
+| 0.7015        | 3.0   | 337  | 0.9337          | 0.7      |
+| 0.4567        | 4.0   | 450  | 0.8595          | 0.79     |
+| 0.2323        | 4.98  | 560  | 0.6152          | 0.84     |
 ### Framework versions