This can be done in parallel to #4 , in order to try to find a good neural network architecture @melli1992