TrainFastImputation function

Learn from the training data so that later you can fill in missing data