Switched to ffnet for neural network