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

Change last layers units to the number of bases

This commit is contained in:
coolneng 2021-07-06 18:01:06 +02:00
parent fba3c5318b
commit ab4a098289
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/model.py
View File

@ -34,7 +34,7 @@ def build_model(hyperparams) -> Model:
units=64, activation="relu", kernel_regularizer=l2(hyperparams.l2_rate) units=64, activation="relu", kernel_regularizer=l2(hyperparams.l2_rate)
), ),
Dropout(rate=0.3), Dropout(rate=0.3),
Dense(units=32, activation="softmax"), Dense(units=len(BASES), activation="softmax"),
] ]
) )
model.compile( model.compile(