plotting of training data too