/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 vote106 views2 answers
Pandas read_csv column names
Python
-1 vote111 views1 answer
ImportError: cannot import name 'FieldDoesNotExist' from 'django.db.models.fields' (C:\Users\Admin\AppData\Local\Programs\Python\Python39\lib\site-packages\django\db\models\fields\__init__.py)
Python
11 votes106 views4 answers
Python execute shell command and get output
Python
0 vote125 views1 answer
Error warning tkinter
Python
5 votes116 views5 answers
Python string format
Python
3 votes146 views6 answers
Django login
Python Django
0 vote111 views2 answers
Python error: command 'x86_64-linux-gnu-gcc' failed with exit status 1
Python
1 vote112 views1 answer
How many data types are specified to numeric values in python
Python
12 votes130 views6 answers
Prime number python program
Python
0 vote150 views1 answer
Django template child data in nested loop
Python Django