This gives us augment support, with the safety of not predicting on the training set
This gives us augment support, with the safety of not predicting on the training set