Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Devanshchowdhury2212/6fd92dc47e182c412729495ff1a02e84 to your computer and use it in GitHub Desktop.
Save Devanshchowdhury2212/6fd92dc47e182c412729495ff1a02e84 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 ='path',y_col='label',target_size=(288,384),batch_size=48,shuffle=True)
valid_generator= datagen_scale.flow_from_dataframe(val,x_col ='path',y_col='label',target_size=(288,384),shuffle=False)
test_generator = datagen_scale.flow_from_dataframe(test,x_col ='path',y_col='label',target_size=(288,384),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