/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 vote357 views2 answers
Python selenium switch to window
Python
-1 vote382 views2 answers
Sequence in python
Python
6 votes370 views2 answers
How to change background of tkinter window
Python
0 vote353 views1 answer
Matplotlib add text to axes
Python
1 vote363 views1 answer
Tkinter center frame
Python
0 vote390 views1 answer
Backticks equivalent python template
Python
0 vote413 views1 answer
Super().__init__(pos, model) in python
Python
0 vote343 views1 answer
Pd.to_datetime(df 'date' errors='coerce')
Python
2 votes373 views2 answers
Django search
Python Django
0 vote357 views1 answer
PHP echo multiple lines example Using concatenation assignment operator
Python