Downloads · 30 days
0
23f2000254-t12026-genre-classifier
pk1308/23f2000254-t12026-genre-classifier
23f2000254-t12026-genre-classifier is a audio classification model from pk1308. Use it for the audio classification task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Hierarchical Vision Transformer for music genre classification using Mel spectrogram and CQT features.
Access
Public
Updated Feb 25, 2026
Repo size
237 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt237 MB · 100%
At a glance
- Task
- Audio Classification
- License
- mit
- Access
- Public
- Created
- Feb 25, 2026
- Updated
- Feb 25, 2026
- SHA
- 5d05499c
Datasets
Model card
From the Hugging Face model README
Music Genre Classifier - 23f2000254-t12026
Hierarchical Vision Transformer for music genre classification using Mel spectrogram and CQT features.
Model Architecture
- Type: Hierarchical Vision Transformer (ViT)
- Input: 7-second audio segments with 2-second overlap (6 segments)
- Features:
- Mel spectrogram (128 bins)
- Constant-Q Transform (84 bins)
- Encoder: Factorized attention with 256 embed dim, 8 heads, 4 layers
- Temporal: Transformer encoder with CLS token aggregation
Training Details
- Dataset: 10,000 synthetic mashups (GTZAN-based)
- Classes: 10 genres (blues, classical, country, disco, hiphop, jazz, metal, pop, reggae, rock)
- Best Val F1: 0.763
- Best Val Accuracy: 76.6%
Files
best_model.pt: Model checkpoint with weights, config, and genre mappinghistory.json: Training history
- Task
- Audio Classification
- License
- mit
- Languages
- en
- Created
- Feb 25, 2026
- Updated
- Feb 25, 2026