我如何从train_generator提取x_train和y_train?

时间:2020-01-10 13:28:04

标签: python-3.x machine-learning deep-learning cross-validation

在我的CNN模型中,我想从train_generator中提取X_trainy_train。我想使用集成学习,打包和增强来评估模型。主要的挑战是如何使用python语言从X_train中提取y_traintrain_generator

 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 :(得分:0)

好吧,首先,您没有编写声明此train_generator的代码。

由于它似乎是keras的生成器,因此应该通过遍历train_generator来访问X_train和y_train。

这意味着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)