Skip to content

Instantly share code, notes, and snippets.

@AmrutaKoshe

AmrutaKoshe/comp_fit.py

Created July 8, 2021 07:53
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/ac91fb9a5116a5f9aa45998257418ea1 to your computer and use it in GitHub Desktop.
Save AmrutaKoshe/ac91fb9a5116a5f9aa45998257418ea1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
comp_fit.py
model.compile(optimizer='adam',
loss='categorical_crossentropy',
metrics=['accuracy'])
history = model.fit(train_generator, validation_data=valid_generator,
steps_per_epoch=train_generator.n//train_generator.batch_size,
validation_steps=valid_generator.n//valid_generator.batch_size,
epochs=10)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment