Update README.md

README.md

@@ -2,7 +2,7 @@
 tags:
 - generated_from_trainer
 datasets:
-- gtzan
+- marsyas/gtzan
 metrics:
 - accuracy
 model-index:
@@ -20,7 +20,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.77
+      value: 0.88
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -28,10 +28,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distilhubert-finetuned-gtzan
 
-This model was trained from scratch on the gtzan dataset.
+This model is a fine-tuned version of ntu-spml/distilhubert on the GTZAN dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.7854
-- Accuracy: 0.77
+- Loss: 0.76
+- Accuracy: 0.88
 
 ## Model description
 
@@ -58,7 +58,8 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
 - num_epochs: 15
-
+- .train_test_split(seed=2024, shuffle=True, test_size=0.1)
+- 
 ### Training results
 
 | Training Loss | Epoch | Step | Accuracy | Validation Loss |