Skip to content

BibuTahseen/AI_Composer

 
 

Repository files navigation

Overview

A project that trains a LSTM recurrent neural network over a dataset of MIDI files. More information can be found on the writeup about this project. This the code for 'Build an AI Composer' on Youtube

Dependencies

Use pip to install any missing dependencies

Basic Usage

  1. mkdir data && mkdir models
  2. run 'python main.py'. This will collect the data, create the chord mapping file in data/nottingham.pickle, and train the model
  3. Run python rnn_sample.py --config_file new_config_file.config to generate a new MIDI song.

Give it 1-2 hours to train on your local machine, then generate the new song. You don't have to wait for it to finish, just wait until you see the 'saving model' message in terminal. In a future video, I'll talk about how to easily setup cloud GPU training. Likely using www.fomoro.com

Credits

Credit for the vast majority of code here goes to Yoav Zimmerman. I've merely created a wrapper around all of the important functions to get people started.

About

AI Composer for Machine Learning for Hackers #2

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%