Split PCA function + Setting train and test matrices