get_partition_indices function

Select indices to partition the data into training & testing sets.