我正在使用flow_from_directory()
和fit_generator
在我的深度学习模型中,我想使用交叉验证方法来训练 CNN 模型。
datagen = ImageDataGenerator(rotation_range=15,width_shift_range=0.2,
height_shift_range=0.2,shear_range=0.2,
zoom_range=0.2,horizontal_flip=True,
fill_mode='nearest')
image_size = (224, 224)
batch = 32
train_generator = datagen.flow_from_directory(train_data,
target_size=image_size,
batch_size=batch,
classes= classes_array)
我找到了这个Youtube 视频和这个教程,但它没有用flow_from_directory()
。
fit_generator
您知道在使用Keras时如何使用 k-fold 交叉验证flow_from_directory()
吗?