Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
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