partition_cv_strat function

Partition the data for a stratified (non-spatial) cross-validation