Advertisement
Guest User

Untitled

a guest
Mar 22nd, 2019
87
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.38 KB | None | 0 0
  1. # initialize our initial learning rate, # of epochs to train for,
  2. # and batch size
  3. INIT_LR = 0.01
  4. EPOCHS = 75
  5. BS = 32
  6.  
  7. # initialize the model and optimizer (you'll want to use
  8. # binary_crossentropy for 2-class classification)
  9. print("[INFO] training network...")
  10. opt = SGD(lr=INIT_LR, decay=INIT_LR / EPOCHS)
  11. model.compile(loss="categorical_crossentropy", optimizer=opt,
  12. metrics=["accuracy"])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement