sklearn的K折交叉验证函数KFold使用

K折交叉验证时使用：

KFold（n_split, shuffle, random_state）

参数：n_split:要划分的折数

shuffle: 每次都进行shuffle，测试集中折数的总和就是训练集的个数

random_state:随机状态

 from sklearn.model_selection import KFold
 kf = KFold(5, True, 10)
 
 X, Y = loda_data('./data.txt')
 for train_index, test_index in kf.split(X):
     print('训练集:{}'.format(train_index)
     print('测试集:{}'.format(test_index)

• 点赞
• 收藏
• 分享
• • 文章举报

隐者之王 发布了23 篇原创文章 · 获赞 6 · 访问量 4613 私信 关注

标签：index,random,shuffle,验证,format,KFold,split,sklearn
来源： https://blog.csdn.net/qq_34211618/article/details/104097498