Cross Validation Code In Python For Creating Variable Size