Skip to content

Instantly share code, notes, and snippets.

@hadifar
Created November 18, 2018 08:40
Show Gist options
  • Save hadifar/36f0056390a706539e9df4a2106fd9a5 to your computer and use it in GitHub Desktop.
Save hadifar/36f0056390a706539e9df4a2106fd9a5 to your computer and use it in GitHub Desktop.
model = Sequential()
model.add(Conv2D(32, kernel_size=(3, 3), activation='relu', input_shape=input_shape))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(num_classes, activation='softmax'))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment