create_train_test_indices function

Create train and test splits to be used in the cross_validation function