Change the NN_train load batch X,Y code.