Skip to content

Instantly share code, notes, and snippets.

@AmrutaKoshe

AmrutaKoshe/train_val.py

Created July 8, 2021 07:50
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 AmrutaKoshe/c5f1bba64c1eda05f306e9f362f65380 to your computer and use it in GitHub Desktop.
Save AmrutaKoshe/c5f1bba64c1eda05f306e9f362f65380 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train_val.py
train_generator = img_datagen.flow_from_directory(directory,
shuffle=True,
batch_size=32,
subset='training',
target_size=(100, 100))
valid_generator = img_datagen.flow_from_directory(directory,
shuffle=True,
batch_size=16,
subset='validation',
target_size=(100, 100))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment