How to Create a Partitioned Custom Dataset