Added to distinguish between predictions and training data set
Rectified training and prediction data conflict