Ну, во-первых, вы не написали кусок кода, объявляющий этот train_generator.
Так как это похоже на генератор keras, вы должны обращаться к X_train и y_train, просматривая train_generator .
Это означает, что train_generator [0] даст вам первую партию пар X_train / y_train.
x_train = []
y_train = []
for x, y in train_generator:
x_train.append(x)
y_train.append(y)