Skip to content

Instantly share code, notes, and snippets.

@chao1224

chao1224/data_split_based_on_index

Created October 19, 2018 20:50
Show Gist options
  • Save chao1224/699feb30fd2cde719ee5523da9f24859 to your computer and use it in GitHub Desktop.
Save chao1224/699feb30fd2cde719ee5523da9f24859 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
data_split_based_on_index
from sklearn.model_selection import KFold
N = 3867
kf = KFold(n_splits=10, shuffle=True)
for _, test_index in kf.split(range(N)):
print(len(test_index))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment