.
ImageDataGenerator(rescale=1./255) # Setup the train and test directories train_dir = "pizza_steak/train/" test_dir = "pizza_steak/test/" # Import data from directories and turn it into batches train_data = train_datagen.flow_from_directory(train_dir, batch_size=32, # number of images to process at