Разделить данные с группами в Python - PullRequest
0 голосов
/ 12 сентября 2018

Я хочу разделить некоторые данные в 3 раза, но без разделения элементов с одинаковым идентификатором.

x=[2,4,5,3,5,6,7,8,8,3,5,2,6,7,8,4]
y=["a","b","c","a","d","f","c","a","a","d","v","f","a","d","r","h"]
groups=[1,1,2,2,3,3,4,4,5,5,6,6,7,7,8,8]

Я пробовал следующий код, но с помощью этой строки я разделяю группы:

folds_x=np.array_split(x, 3)
folds_y=np.array_split(y, 3)

Я попробовал следующую команду, но я хочу создать только 10 сгибов, а не сгибов по K-1:

generator = GroupKFold(n_splits=10).split(x, y, groups)
for train_index,test_index in generator:
   print(len(train_index))

Как я могу это сделать?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...