将数据集分成k分
k-1份用来训练,1份用来测试,遍历所有数据
程序:
kfold = StratifiedKFold(n_splits=number, random_state=1).split(x, y)
for k, (train, test) in enumerate(kfold):
#k表示k折,train和test是索引,在训练集和测试集时会保证每次标签值分配均匀,避免出现一个训练集或测试集内一个标签只出现一次
将数据集分成k分
k-1份用来训练,1份用来测试,遍历所有数据
程序:
kfold = StratifiedKFold(n_splits=number, random_state=1).split(x, y)
for k, (train, test) in enumerate(kfold):
#k表示k折,train和test是索引,在训练集和测试集时会保证每次标签值分配均匀,避免出现一个训练集或测试集内一个标签只出现一次