/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

2 votes334 views2 answers
Python include file
Python
0 vote337 views1 answer
Python limit round with 2 decimal
Python
0 vote366 views1 answer
Python insert parent directory into sys path for import file purpose
Python
0 vote325 views1 answer
Python check if more than 1 is true
Python
0 vote316 views1 answer
Python coding for y, you will also display a “bar” of ‘X’ characters to represent the number. For example, the prime number 2 would be represented as “X 2”.
Python
0 vote369 views1 answer
Matplotlib force scientific notation and define exponent
Python
0 vote431 views3 answers
Pysftp get-r
Python
0 vote355 views1 answer
Custom jupyter notebook
Python
0 vote333 views1 answer
Get duplicate and remove but keep last in python df
Python
0 vote328 views1 answer
Networkx node attribute from a dataframe
Python