Skip to content

Instantly share code, notes, and snippets.

@aniruddha27

aniruddha27/model_building.py

Last active August 8, 2020 10:23
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 aniruddha27/b637cb9b44764a6a5b49a74846e96a1c to your computer and use it in GitHub Desktop.
Save aniruddha27/b637cb9b44764a6a5b49a74846e96a1c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_building.py
# Model architechture
model = Sequential()
model.add(Conv2D(32, (3, 3), input_shape=(224, 224, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(32, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Flatten())
model.add(Dense(64, activation='relu'))
model.add(Dropout(0.24))
model.add(Dense(2,activation='softmax'))
model.summary()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment