Configuring the training parameters of the neural network