Tag neural network training