Skip to content

Instantly share code, notes, and snippets.

@d0znpp
Created December 12, 2017 00:46
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 d0znpp/7805cb2907724f1e6c4c189149c61f18 to your computer and use it in GitHub Desktop.
Save d0znpp/7805cb2907724f1e6c4c189149c61f18 to your computer and use it in GitHub Desktop.
class NetManager():
def __init__(self, num_input, num_classes, learning_rate, mnist,
max_step_per_action=5500,
bathc_size=100,
dropout_rate=0.85):
self.num_input = num_input
self.num_classes = num_classes
self.learning_rate = learning_rate
self.mnist = mnist
self.max_step_per_action = max_step_per_action
self.bathc_size = bathc_size
self.dropout_rate = dropout_rate #Dropout after dense layer in CNN
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment