Повторное назначение KFold поезда и теста не работает

Question

У меня есть вектор строки index размер (4,1)
x - это матрица (12,10)
Я хочу установить a = x [index]

Я получил это от RepeatedKFold. Это l oop не работает.

for train_index, test_index in rkf.split(X):
...     print("TRAIN:", train_index, "TEST:", test_index)
...     X_train, X_test = X[train_index], X[test_index]
...     y_train, y_test = y[train_index], y[test_index]