Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # initialize our initial learning rate and # of epochs to train for
- INIT_LR = 0.01
- EPOCHS = 75
- # compile the model using SGD as our optimizer and categorical
- # cross-entropy loss (you'll want to use binary_crossentropy
- # for 2-class classification)
- print("[INFO] training network...")
- opt = SGD(lr=INIT_LR)
- model.compile(loss="categorical_crossentropy", optimizer=opt,metrics=["accuracy"])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement