/Python/How to split image dataset into training and test set keras

How to split image dataset into training and test set keras

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

Source: stackoverflow.com

Similar codes

0 vote205 views1 answer
Python running mean pandas
Python
5 votes199 views4 answers
Prime numbers from 1 to 100 in python
Python
1 vote225 views1 answer
How to python hack 2021 course
Python
0 vote184 views1 answer
Which clause is used to place condition with GROUP BY clause in a table
Python
-1 vote193 views1 answer
Numpy apply function to array
Python
0 vote216 views1 answer
Python jinja2 from string
Python
0 vote184 views2 answers
Connect to mysql database jupyter
Python
12 votes204 views4 answers
Python remove
Python
79 votes188 views6 answers
Python sort dictionary by value
Python
85 votes200 views6 answers
For python
Python