/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
Python fillna with mean in a dataframe
Python
7 votes187 views3 answers
Python argparse choice
Python
0 vote199 views2 answers
IPTC text classification example
Python
1 vote228 views1 answer
Access-Control-Allow-Origin django
Python Django
3 votes275 views2 answers
To extract out only year month from a date column in pandas
Python
1 vote181 views1 answer
For i in a for j in a loop python
Python
29 votes200 views6 answers
Update all pip packages
Python
-2 votes266 views6 answers
Django download
Python Django
0 vote196 views1 answer
Python read excel sheet name
Python
4 votes182 views2 answers
Timeout socket python
Python