add neural network training