Skip to content

zqpie/Predictive-Song-Genre-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Project uses machine learning to classify audio recordings of format .wav into trained generes, based on features.
- Extracts audio features (MFCCs, chroma, etc.) using `librosa`
- Trains a Random Forest classifier to predict genres
- Supports prediction for new `.wav` files


Model trained using data set in the below format:

Data/genres_original/ then all the genres get folders. their name is used for classification. 
or just use this premade set:
https://www.kaggle.com/datasets/andradaolteanu/gtzan-dataset-music-genre-classification/data

To use:
edit the path line on run Generate_feature_set.py to the folder containing the genre folders. then run this file. 
it will generate a .csv feature set that can be used by predict_new_song.py, run this file.

About

This project builds a machine learning system to classify music tracks into genres based on their audio characteristics. Using librosa to mark features such as tempo, zero crossing rate, and special contrast.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages