Neural network with 2 hidden layers and 396 features