stratified k fold
data = array([9,5,6,7,8,3]) target= [1,2,1,2,1,2] stratifiedkfold = StratifiedKFold(n_splits=2) for training, testing in stratifiedkfold.split(data,target): print('training: %s, testing: %s' % (data[training], data[testing]))
Not a member yet? Register now
Are you a member? Login now