| Title | Author |
|---|---|
Music Classification |
Alessandro Stockman |
Comparison between Wav2Vec2 and Whisper fine-tuned for music genre classification
- 1_exploration.ipynb | Analysis of the dataset
- 2_training.ipynb | Training of different architectures
- 3_evaluation.ipynb | Evaluations on the trained models
Tested with python 3.10.6 and the following libraries:
- pandas
- matplotlib
- torch
- datasets
- transformers
- evaluations
- torchaudio
- wandb
- librosa
- evaluate
- tqdm
- scikit-learn
- ffmpeg-python
System dependencies:
- ffmpeg
- sox