Training with 2 hidden layers and 195 features