The project aims at training a model for music categorization. It will accept a music track given in input and will return the category of the music (pop, rock, ...) among the ones the model learnt.
Below is the list of all categories taken into account in this project:
- blues
- classical
- country
- disco
- hiphop
- jazz
- metal
- pop
- reggae
- rock
-
requirements.txt
allows you to install all dependencies to the project. See 'Set up' section in order to install them. -
baseline_model
is the directory containing all python files and notebooks regarding the baseline model. -
Data
folder contains 2csv
files used for training baseline model.
Whether you want to use the notebooks or the python files, you need to resolve all dependencies. You can install them either in your virtual environment or Python base environment with the following command: pip install -r requirements.txt
.
If you want to use notebooks defined in this repository, you can upload them on Google Collab or you can run a Jupyter notebook server locally. You can install Jupyter by following the official documentation.