coolneng
/
locimend
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Build model incrementally

This commit is contained in:
coolneng 2021-06-14 23:32:49 +02:00
parent 19ed847d12
commit d2e5fd0fa3
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 37 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
src/model.py
View File

@ -15,39 +15,44 @@ def build_model() -> Model:
"""
Build the CNN model
"""
model = Sequential(
[
model = Sequential()
model.add(
layers.Conv1D(
filters=16,
kernel_size=5,
activation="relu",
kernel_regularizer=l2(L2),
),
layers.MaxPool1D(pool_size=3, strides=1),
)
)
model.add(layers.MaxPool1D(pool_size=3, strides=1))
model.add(
layers.Conv1D(
filters=16,
kernel_size=3,
activation="relu",
kernel_regularizer=l2(L2),
),
layers.MaxPool1D(pool_size=3, strides=1),
layers.Flatten(),
layers.Dense(
units=16,
activation="relu",
kernel_regularizer=l2(L2),
),
layers.Dropout(rate=0.3),
layers.Dense(
units=16,
activation="relu",
kernel_regularizer=l2(L2),
),
layers.Dropout(rate=0.3),
# FIXME Change output size
layers.Dense(units=len(BASES), activation="softmax"),
]
)
)
model.add(layers.MaxPool1D(pool_size=3, strides=1))
model.add(layers.Flatten())
model.add(
layers.Dense(
units=16,
activation="relu",
kernel_regularizer=l2(L2),
)
)
model.add(layers.Dropout(rate=0.3))
model.add(
layers.Dense(
units=16,
activation="relu",
kernel_regularizer=l2(L2),
)
)
model.add(layers.Dropout(rate=0.3))
# FIXME Change output size
model.add(layers.Dense(units=len(BASES), activation="softmax"))
model.compile(
optimizer=Adam(LEARNING_RATE),
loss=sparse_categorical_crossentropy,