generate train data, test, k_fold_cross_test