train_test function

Split choice data in train and test subset