Add a comment that you are doing random subsets here for the training