Proportional Class Partitioning for Dataset Splitting - Coding problem | OneNoughtOne