Fixed the bug that was reducing the number of neural net outputs.