/Python/Split imagedatagenerator into x_train and y_train

Split imagedatagenerator into x_train and y_train

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

Split imagedatagenerator into x_train and y_train

in python 2:

X_train, y_train = train_generator.next()
X_test, y_test = validation_generator.next()

in python 3:

X_train, y_train = next(train_generator)
X_test, y_test = next(validation_generator)

Similar codes

58 votes207 views6 answers
Python class definition
Python
0 vote124 views2 answers
Django run management command from code
Python
5 votes129 views3 answers
How to save unzipped files in python
Python
0 vote137 views1 answer
Python how to move multiple values from one list to another at once
Python
22 votes122 views5 answers
Python palindrome check
Python
10 votes126 views1 answer
How do i convert a list to a string in python
Python
0 vote131 views1 answer
Matplotlib savefig legend cut off
Python
8 votes121 views1 answer
Append one row to pandas dataframe
Python
3 votes190 views1 answer
Django session expire time
Python Django
0 vote130 views1 answer
Django filter queryset by date
Python