Commit 1e2119f7 by Pamela Osuna

second layer option added

parent 29716a7c
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, Dropout, Flatten
from tensorflow.keras.layers import Conv1D, MaxPooling1D
from tensorflow.keras.layers import Conv1D, MaxPooling1D, LeakyReLU
# models
num_classes = 4
......@@ -25,7 +25,16 @@ def model_architecture(c):
model.add(Dense(num_classes, activation = 'softmax'))
if c == 1:
pass
model.add(Conv1D(16, kernel_size = 3, activation = 'linear', input_shape = (30, 1), padding = 'same'))
model.add(LeakyReLU(alpha = 0.1))
model.add(MaxPooling1D(2, padding = 'same'))
model.add(Conv1D(64,3, activation = 'linear', padding = 'same'))
model.add(LeakyReLU(alpha = 0.1))
model.add(MaxPooling1D(pool_size = 2, padding = 'same'))
model.add(Flatten())
model.add(Dense(128, activation = 'linear'))
model.add(LeakyReLU(alpha = 0.1))
model.add(Dense(num_classes, activation = 'softmax'))
if c == 2:
pass
......
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