Advertisement
Guest User

Untitled

a guest
Mar 19th, 2019
68
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 and # of epochs to train for
  2. INIT_LR = 0.01
  3. EPOCHS = 75
  4.  
  5. # compile the model using SGD as our optimizer and categorical
  6. # cross-entropy loss (you'll want to use binary_crossentropy
  7. # for 2-class classification)
  8. print("[INFO] training network...")
  9. opt = SGD(lr=INIT_LR)
  10. model.compile(loss="categorical_crossentropy", optimizer=opt,metrics=["accuracy"])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement