Train and test set separation.