Made the train/cv/test split an option in neural_network_init instead of hard coded.