diff --git a/38-sliding-window/tf-38.py b/38-sliding-window/tf-38.py index 8139f5c..6b6e0f1 100644 --- a/38-sliding-window/tf-38.py +++ b/38-sliding-window/tf-38.py @@ -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()