A machine learning project that classifies music into various genres using Convolutional Neural Networks (CNN). The project leverages the GTZAN dataset and pre-trained models to predict music genres based on audio features.
Use this Python file to make genre predictions on audio files (MP3 or WAV format).
Note: The required pre-trained model file is already included in this repository.
This script extracts the necessary features from the music files. Before running it, you'll need to download the GTZAN dataset and set the DATASET_PATH variable to point to the correct directory where the dataset is stored.
Once you’ve prepared the dataset (via Dataset_Preparation.py), use this script to train the neural network. It will generate and export the trained model in JSON format.
Below are some visuals that show the accuracy of the network:
