Support training and test split for training table