duault
commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -3,6 +3,14 @@ datasets:
|
|
| 3 |
- lewtun/music_genres_small
|
| 4 |
base_model:
|
| 5 |
- facebook/wav2vec2-large
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
---
|
| 7 |
|
| 8 |
# My Music Genre Classification Model 🎶
|
|
@@ -29,4 +37,4 @@ inputs = feature_extractor(audio, sampling_rate=16000, return_tensors="pt")
|
|
| 29 |
# Make predictions
|
| 30 |
logits = model(**inputs).logits
|
| 31 |
predicted_class = torch.argmax(logits, dim=-1).item()
|
| 32 |
-
print(predicted_class)
|
|
|
|
| 3 |
- lewtun/music_genres_small
|
| 4 |
base_model:
|
| 5 |
- facebook/wav2vec2-large
|
| 6 |
+
metrics:
|
| 7 |
+
- accuracy
|
| 8 |
+
- f1
|
| 9 |
+
tags:
|
| 10 |
+
- audio
|
| 11 |
+
- music
|
| 12 |
+
- classification
|
| 13 |
+
- Wav2Vec2
|
| 14 |
---
|
| 15 |
|
| 16 |
# My Music Genre Classification Model 🎶
|
|
|
|
| 37 |
# Make predictions
|
| 38 |
logits = model(**inputs).logits
|
| 39 |
predicted_class = torch.argmax(logits, dim=-1).item()
|
| 40 |
+
print(predicted_class)
|