stratifiedgroupkfold example
data = array([9,5,6,7,8,3]) target= [1,2,1,2,1,1] test_groups=[1,1,2,2,1,2] stratifiedgroupkfold = StratifiedGroupKFold(n_splits=2) for training, testing in stratifiedgroupkfold.split(data,target,groups=test_groups): print('training: %s, testing: %s' % (data[training], data[testing]))