Breaking News: Grepper is joining You.com. Read the official announcement!

How to do train test split in keras Imagedatagenerator

Darkstar answered on September 4, 2021 Popularity 6/10 Helpfulness 5/10

answer How to do train test split in keras Imagedatagenerator

How to do train test split in keras Imagedatagenerator

Comment

train_datagen = ImageDataGenerator(rescale=1./255,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True,
    validation_split=0.2) # set validation split

train_generator = train_datagen.flow_from_directory(
    train_data_dir,
    target_size=(img_height, img_width),
    batch_size=batch_size,
    class_mode='binary',
    subset='training') # set as training data

validation_generator = train_datagen.flow_from_directory(
    train_data_dir, # same directory as training data
    target_size=(img_height, img_width),
    batch_size=batch_size,
    class_mode='binary',
    subset='validation') # set as validation data

model.fit_generator(
    train_generator,
    steps_per_epoch = train_generator.samples // batch_size,
    validation_data = validation_generator, 
    validation_steps = validation_generator.samples // batch_size,
    epochs = nb_epochs)

Popularity 6/10 Helpfulness 5/10 Language python

Source: stackoverflow.com

Tags: imagedatagenerator keras python

Link to this answer
Share Copy Link

Contributed on Sep 04 2021

Darkstar

0 Answers Avg Quality 2/10

How to do train test split in keras Imagedatagenerator

Contents

More Related Answers

How to do train test split in keras Imagedatagenerator

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.