Added the learning counterpart of 37

This commit is contained in:
Crista Lopes
2020-01-02 09:56:35 -08:00
parent ac3d348e45
commit e8b48906e2
2 changed files with 107 additions and 4 deletions

View File

@@ -69,10 +69,9 @@ def normalization_layer_set_weights(n_layer):
def build_model():
# Normalize characters using a dense layer
model = Sequential()
dense_layer = Dense(LINE_SIZE*INPUT_VOCAB_SIZE,
input_shape=(LINE_SIZE*INPUT_VOCAB_SIZE,),
activation='softmax')
model.add(dense_layer)
model.add(Dense(LINE_SIZE*INPUT_VOCAB_SIZE,
input_shape=(LINE_SIZE*INPUT_VOCAB_SIZE,),
activation='sigmoid'))
return model
model = build_model()