CNN that classifies different genres of music
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
.gitignore
LICENSE
README.md
cnn.ipynb

README.md

WyGen: music genre classifier

Convolutional Neural Network that classifies different genres of music

Based on the AlexNet architecture, the network consists of 3 convolutional layers, and 2 fully-connected layers, classifying music into 10 different genres (Blues, Classical, Country, Disco, Hip-Hop, Jazz, Metal, Pop, Rock, and Reggae). With an accuracy of 63.7% it is not state-of-the-art(6 genres:accuracy of 84%), but it performs well above the 10% chance.

Future Changes

  • Adding an LSTM (or another derivative recurrent neural net) layer that helps capture the sequential nature of music.
  • Add 1-2 more convolutional layers to capture even lower level features.
  • Train and test on more data.