Skip to content

Instantly share code, notes, and snippets.

@ferhat00

ferhat00/transfer17.py

Created October 31, 2018 08:36
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save ferhat00/b48b6daa4b7585b40515f260a6093370 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save ferhat00/b48b6daa4b7585b40515f260a6093370 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
transfer17.py
model.compile(optimizer='Adam',loss='categorical_crossentropy',metrics=['accuracy'])
# Adam optimizer
# loss function will be categorical cross entropy
# evaluation metric will be accuracy
step_size_train=train_generator.n//train_generator.batch_size
model.fit_generator(generator=train_generator,
steps_per_epoch=step_size_train,
epochs=10)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment