abhatem / genre_classification

Description

hobby project I made experimenting with basic neural network

Dataset

GTZAN dataset (https://www.kaggle.com/datasets/andradaolteanu/gtzan-dataset-music-genre-classification)

extract the Data folder next to the genre_classification.ipynb notebook

Usage

after extracting the Data folder, run the following to install pytorch cuda

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

then

pip install -r requirements.txt

then run the genre_classification.ipynb notebook