Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/model_building.py

Created September 5, 2019 06:49
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save NMZivkovic/dc53ff182158e1e47088c77f04060bcd to your computer and use it in GitHub Desktop.
Save NMZivkovic/dc53ff182158e1e47088c77f04060bcd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_building.py
model = Sequential()
model.add(Conv2D(32, 3, activation='relu', input_shape=(28,28, 1)))
model.add(Conv2D(64, 3, activation='relu'))
model.add(Conv2D(128, 3, activation='relu'))
model.add(MaxPooling2D(2, 2))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dense(10, activation='softmax'))
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment