Skip to content

BartoszDorobek/Music-genre-recognition

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Music genre recognition with AI

1. Machine learning models

Datasource was a tabular representation with extracted audio features:

raw_data_2

I chose a Bayesian search method to find the best hyperparameters for each algorithm.

modele_acc

Results were compared for multiple ML algorithms and for two different audio sample lengths.

image

The best accuracy (91,24%) was achieved by the LightGBM model trained on 3-second audio.

macierz_pomylek_3

2. Convolutional Neutral Network

For the second part of the project I used Mel-spectrogram representation of audio.

image

I compared many CNN architectures to choose the best one, which I present below and which achieved an accuracy level of 85,06%.

image

train_val_acc

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published