sklearn的K折交叉验证函数KFold使用
K折交叉验证时使用:
KFold(n_split, shuffle, random_state)
参数:n_split:要划分的折数
shuffle: 每次都进行shuffle,测试集中折数的总和就是训练集的个数
random_state:随机状态
1 from sklearn.model_selection import KFold 2 kf = KFold(5, True, 10) 3 4 X, Y = loda_data('./data.txt') 5 for train_index, test_index in kf.split(X): 6 print('训练集:{}'.format(train_index) 7 print('测试集:{}'.format(test_index)
时刻记着自己要成为什么样的人!
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步