kfold生成训练集_weixin_45955767的博客

生成fold值

fold 折数

df_label 导入的csv文件

fold_lst = (len(df_label)//fold)*[i for i in range(fold)]
r=random.random
random.seed(random_seed)
random.shuffle(fold_lst,random=r)
df_label['fold'] = fold_lstfold_ls