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