In order to replicate figures, one needs to set hyperparameters at the top of the stub.py file to desired values, save the file and 
execute:

> python stub.py

To generate random policy figure:

> python stub_random.py

Dependecies: numpy, matplotlib, keras VERSION 0.3.2 - IMPORTANT TO HAVE THE RIGHT VERSION OF KERAS, which can be installed with pip install keras=0.3.2