Splitting training set based on ids provided in file