Assign to training set