Fix normalization to support larger test matrices
Classification: prepare the dataset to predict