Change last layers units to the number of bases

2021-07-06 18:01:06 +02:00 · 2021-07-06 18:01:06 +02:00 · ab4a098289
parent fba3c5318b
commit ab4a098289
1 changed files with 1 additions and 1 deletions
--- 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(