The data generation is fast enough but I can replace it with a Theano function if that is preferable. The purpose of being able to save and load the data is to allow for plotting it.