Skip to content

Instantly share code, notes, and snippets.

@amankharwal
Created December 27, 2020 14:10
Show Gist options
  • Star 2 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save amankharwal/aa8e1ec17a3a9c5fa4ed3cc3d637a65f to your computer and use it in GitHub Desktop.
Save amankharwal/aa8e1ec17a3a9c5fa4ed3cc3d637a65f to your computer and use it in GitHub Desktop.
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dropout, Dense
cnn = Sequential([Conv2D(filters=100, kernel_size=(3,3),
activation='relu'),
MaxPooling2D(pool_size=(2,2)),
Conv2D(filters=100, kernel_size=(3,3),
activation='relu'),
MaxPooling2D(pool_size=(2,2)),
Flatten(),
Dropout(0.5),
Dense(50),
Dense(35),
Dense(2)])
cnn.compile(optimizer='adam', loss='binary_crossentropy', metrics=['acc'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment