PCA to speed up model training