Getting initial numbers on training and test set