/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 vote179 views1 answer
Break input loop
Python
1 vote190 views3 answers
Recursive function
Python
25 votes132 views7 answers
Remove space from start and end of string in python
Python
0 vote217 views1 answer
Cbv uk django
Python Django
0 vote218 views1 answer
Flask No application found. Either work inside a view function or push an application context
Python Flask
-4 votes178 views5 answers
Python 3.6 release date
Python
0 vote186 views1 answer
My pygame window wont stay open
Python
0 vote167 views1 answer
How to get python to write to 100
Python
0 vote184 views1 answer
Numpy convolution stride tricks
Python
1 vote206 views1 answer
Django queryset first element
Python Django