Fix k-fold CV

In k-fold cross-validation, the single subsample is used as the validation
data for testing the model, and the remaining k − 1 subsamples are used
for training.
Merge pull request #4 from BigCrunsh/fix_cv

Fix k-fold CV