Skip to content
This repository has been archived by the owner on Aug 25, 2020. It is now read-only.

Latest commit

 

History

History
38 lines (32 loc) · 680 Bytes

README.md

File metadata and controls

38 lines (32 loc) · 680 Bytes
Raw

Music Genre Classification using Logistic Regression

Dataset Heirarchy

gtzan_dataset
	\blues
	\classical
	\country
	\disco
	\hiphop
	\jazz
	\metal
	\pop
	\reggae
	\rock

Run order of files:

python3 wav_convert.py

Converts music files to the needed .wav format and store into ./converted

python3 wav_extract_fft.py

Extracts fft values and saves numpy files to ./fft_extracted

python3 log_reg.py fft_extracted/

Applies Logistic Regression to train and classify; gives prediction accuracy, saves confusion matrix image and model

python3 test.py ~/path/to/your/song/file

Loads saved model to predict genre of new file