barryridge / AI_Composer

AI Composer for Machine Learning for Hackers #2

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

• Numpy (http://www.numpy.org/)
• Tensorflow (https://github.com/tensorflow/tensorflow)
• Python Midi (https://github.com/vishnubob/python-midi.git)
• Mingus (https://github.com/bspaans/python-mingus)

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.