Data Partitioning with Cross-Validation Splits - Coding problem | OneNoughtOne