Add stratified_cross_validate_partition