Normalize based on mean and std of training set only!
Normalize based on mean and std of training set only!