Split training set and data set