| | --- |
| | metrics: |
| | - accuracy |
| | pipeline_tag: audio-classification |
| | tags: |
| | - music |
| | --- |
| | This model is finetuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) for music genre classification. |
| | # This model was finetuned to distinguish between 18 music genres including: |
| | - Blues |
| | - Classical music |
| | - Country music |
| | - Drum & Bass |
| | - Dubstep |
| | - Folk |
| | - Future Bass |
| | - Hardstyle |
| | - House |
| | - Jazz |
| | - Latin music |
| | - Metal |
| | - Pop |
| | - R&B, Soul |
| | - Rap |
| | - Reagge |
| | - Rock |
| | - Trap |
| |
|
| | # Training hyperparameters |
| | The following hyperparameters were used during training: |
| | - metric_for_best_model = accuracy |
| | - learning_rate = 5e-5 |
| | - seed = 42 |
| | - per_device_train_batch_size = 4 |
| | - per_device_eval_batch_size = 4 |
| | - gradient_accumulation_steps = 1 |
| | - warmup_ratio = 0.1 |
| | - fp16 = True |
| | - adam_epsilon = 1e-08 |
| | - adam_beta1 = 0.9 |
| | - adam_beta2 = 0.999 |