plot the distribution on training data