Skip to content

Instantly share code, notes, and snippets.

@fuwiak

fuwiak/model_cnn.py

Created July 25, 2020 10:53
Show Gist options
  • Save fuwiak/ce33056afb135b1ea0ebd994f1c59acc to your computer and use it in GitHub Desktop.
Save fuwiak/ce33056afb135b1ea0ebd994f1c59acc to your computer and use it in GitHub Desktop.
Download ZIP
Raw
model_cnn.py
num_of_class = 7;
model = Sequential()
model.add(Conv2D(10, (2, 2), activation='softmax', input_shape=(48,48,1)))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(64, (5, 5), activation='softmax'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Flatten())
model.add(Dense(1000, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(500, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(250, activation='relu'))
model.add(Dense(num_of_class, activation='softmax'))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment