add k-fold cv support