def split_kfold_c(y):
skf = StratifiedKFold(5)
ilst = []
for tri, tei in skf.split(np.zeros(len(y)), y):
ilst.append((tri, tei))
return ilst
微信
公众号
扫码关注公众号