k_fold function

K-fold sampling