Как я могу извлечь x_train и y_train из train_generator? - PullRequest
0 голосов
/ 10 января 2020

В моей модели CNN я хочу извлечь X_train и y_train из train_generator. Я хочу использовать ансамблевое обучение, сумку и повышение, чтобы оценить модель. Основная проблема заключается в том, как я могу извлечь X_train и y_train из train_generator, используя python язык.

 history=model.fit_generator(train_generator, 
                             steps_per_epoch=num_of_train_samples // batch_size,
                             epochs=10, validation_data=validation_generator, 
                             validation_steps=num_of_val_samples // batch_size, 
                             callbacks=callbacks)

1 Ответ

0 голосов
/ 10 января 2020

Ну, во-первых, вы не написали кусок кода, объявляющий этот 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)
...