Как разделить набор данных тренировочного образа на X_train и y_train?

Question

Я пытаюсь разделить набор данных тренировочного образа на X_train и y_train, но я все еще не знаю, как это сделать.

train_path = '/content/drive/My Drive/CBD Robotics course/Assignment 08/fruits/fruits-360//Training'
test_path = '/content/drive/My Drive/CBD Robotics course/Assignment 08/fruits/fruits-360//Test'

from keras.preprocessing.image import ImageDataGenerator
# create a data generator
datagen = ImageDataGenerator()

# load and iterate training dataset
train_it = datagen.flow_from_directory(directory=train_path, target_size=(100, 100), color_mode="rgb",  class_mode='categorical', batch_size=64)
# load and iterate test dataset
test_it = datagen.flow_from_directory(directory=test_path, target_size=(100, 100), color_mode="rgb", class_mode='categorical', batch_size=64)

Я надеюсь, что получу что-то вроде примера кода, ноиз моих данных:

(X_train, y_train), (X_test, y_test) = mnist.load_data()

Как преобразовать переменные train_it и test_it в массив numpy и получить части ввода и метки?