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)

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM