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

时间:2021-12-25 05:55:52

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)