相关疑难解决方法(0)

使用ImageDataGenerator时,Keras分割列车测试集

我有一个目录,其中包含图像的子文件夹(根据标签).我想在Keras中使用ImageDataGenerator时将这些数据拆分为train和test set.尽管keras中的model.fit()具有用于指定拆分的参数validation_split,但我找不到与model.fit_generator()相同的参数.怎么做 ?

train_datagen = ImageDataGenerator(rescale=1./255,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True)

train_generator = train_datagen.flow_from_directory(
    train_data_dir,
    target_size=(img_width, img_height),
    batch_size=32,
    class_mode='binary')

model.fit_generator(
    train_generator,
    samples_per_epoch=nb_train_samples,
    nb_epoch=nb_epoch,
    validation_data=??,
    nb_val_samples=nb_validation_samples)
Run Code Online (Sandbox Code Playgroud)

我没有单独的验证数据目录,需要将其从训练数据中分离出来

keras train-test-split

38
推荐指数
6
解决办法
2万
查看次数

标签 统计

keras ×1

train-test-split ×1