Minor clean up

This commit is contained in:
Crista Lopes
2020-01-02 09:57:02 -08:00
parent e8b48906e2
commit a81889f2bb

View File

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