k折交叉验证KFold()函数的使用

时间:2020-03-18 22:20:56   收藏:0   阅读:768

KFold(n_split, shuffle, random_state)

  参数:n_splits:要划分的折数

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

     random_state:随机状态

 

from sklearn.model_selection import KFold

kf = KFold(n_splits=3,random_state=1)

for train, test in kf.split(titanic):

  titanic为X,即要使用的数据集

原文:https://www.cnblogs.com/zjkbk/p/12520350.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!