Update README.md

README.md

@@ -19,19 +19,23 @@ pipeline_tag: audio-classification
 
 # wav2vec2-base-Drum_Kit_Sounds
 
-This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the audiofolder dataset.
+This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base).
+
 It achieves the following results on the evaluation set:
 - Loss: 1.0887
 - Accuracy: 0.7812
-- Weighted f1: 0.7692
-- Micro f1: 0.7812
-- Macro f1: 0.7845
-- Weighted recall: 0.7812
-- Micro recall: 0.7812
-- Macro recall: 0.8187
-- Weighted precision: 0.8717
-- Micro precision: 0.7812
-- Macro precision: 0.8534
+- F1
+  - Weighted: 0.7692
+  - Micro: 0.7812
+  - Macro: 0.7845
+- Recall
+  - Weighted: 0.7812
+  - Micro: 0.7812
+  - Macro: 0.8187
+- Precision
+  - Weighted: 0.8717
+  - Micro: 0.7812
+  - Macro: 0.8534
 
 ## Model description
 
@@ -63,7 +67,7 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy | Weighted f1 | Micro f1 | Macro f1 | Weighted recall | Micro recall | Macro recall | Weighted precision | Micro precision | Macro precision |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | Weighted F1 | Micro F1 | Macro F1 | Weighted Recall | Micro Recall | Macro Recall | Weighted Precision | Micro Precision | Macro Precision |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:-----------:|:--------:|:--------:|:---------------:|:------------:|:------------:|:------------------:|:---------------:|:---------------:|
 | 1.3743        | 1.0   | 4    | 1.3632          | 0.5625   | 0.5801      | 0.5625   | 0.5678   | 0.5625          | 0.5625       | 0.5670       | 0.6786             | 0.5625          | 0.6429          |
 | 1.3074        | 2.0   | 8    | 1.3149          | 0.3438   | 0.2567      | 0.3438   | 0.2696   | 0.3438          | 0.3438       | 0.375        | 0.3067             | 0.3438          | 0.3148          |