yoavz / music_rnn

Music Language Modeling with Recurrent Neural Networks
http://yoavz.com/music_rnn
197 stars 38 forks source link

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 or the final report written. Warning: Some parts of this codebase are unfinished.

Dependencies

Basic Usage

  1. Run ./install.sh to create conda env, install dependencies and download data
  2. source activate music_rnn to activate the conda environment
  3. Run python nottingham_util.py to generate the sequences and chord mapping file to data/nottingham.pickle
  4. Run python rnn.py --run_name YOUR_RUN_NAME_HERE to start training the model. Use the grid object in rnn.py to edit hyperparameter configurations.
  5. source deactivate to deactivate the conda environment