Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save Devanshchowdhury2212/56be6585094ff4236d3f064da794d4e6 to your computer and use it in GitHub Desktop.
Save Devanshchowdhury2212/56be6585094ff4236d3f064da794d4e6 to your computer and use it in GitHub Desktop.
datagen_scale = keras.preprocessing.image.ImageDataGenerator(rescale=1/255)
train_generator= datagen_scale.flow_from_dataframe(train,x_col ='filepaths',y_col='labels',target_size=(224,224),batch_size=64,shuffle=True)
valid_generator= datagen_scale.flow_from_dataframe(val,x_col ='filepaths',y_col='labels',target_size=(224,224),shuffle=False)
test_generator = datagen_scale.flow_from_dataframe(test,x_col ='filepaths',y_col='labels',target_size=(224,224),shuffle=False)
model.evaluate(train_generator)
model.evaluate(valid_generator)
model.evaluate(test_generator)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment