Commit db5a9e46 by Pamela Osuna

2 remaining layer architecture added

parent 2dff1016
......@@ -37,10 +37,29 @@ def model_architecture(c):
model.add(Dense(num_classes, activation = 'softmax'))
if c == 2:
pass
model.add(Conv1D(filters=64, kernel_size=3, activation='relu', input_shape=(30,1)))
model.add(Conv1D(filters=64, kernel_size=3, activation='relu'))
model.add(Dropout(0.5))
model.add(MaxPooling1D(pool_size=2))
model.add(Flatten())
model.add(Dense(100, activation='relu'))
model.add(Dense(num_classes, activation='softmax'))
if c == 3:
pass
model.add(Conv1D(32, kernel_size = 3, activation='relu', input_shape=(30,1)))
model.add(Conv1D(32, 3, activation='relu'))
model.add(MaxPooling1D(pool_size=2))
model.add(Dropout(0.25))
model.add(Conv1D(64, 3, activation='relu'))
model.add(Conv1D(64, 3, activation='relu'))
model.add(MaxPooling1D(pool_size=2))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(256, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(num_classes, activation='softmax'))
return model
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment