From ab4a098289ebae2942348fe0d1ecf165efd5ca06 Mon Sep 17 00:00:00 2001 From: coolneng Date: Tue, 6 Jul 2021 18:01:06 +0200 Subject: [PATCH] Change last layers units to the number of bases --- src/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/model.py b/src/model.py index e113fb0..39f7ab4 100644 --- a/src/model.py +++ b/src/model.py @@ -34,7 +34,7 @@ def build_model(hyperparams) -> Model: units=64, activation="relu", kernel_regularizer=l2(hyperparams.l2_rate) ), Dropout(rate=0.3), - Dense(units=32, activation="softmax"), + Dense(units=len(BASES), activation="softmax"), ] ) model.compile(