From 40c292a4f7a4d8001997a84db64be17bbce36f46 Mon Sep 17 00:00:00 2001 From: Crista Lopes Date: Sat, 28 Dec 2019 18:24:52 -0800 Subject: [PATCH] Make 35 even more like 36 --- 35-dense-shallow-under-control/tf-35.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/35-dense-shallow-under-control/tf-35.py b/35-dense-shallow-under-control/tf-35.py index c8668bd..5eddb38 100644 --- a/35-dense-shallow-under-control/tf-35.py +++ b/35-dense-shallow-under-control/tf-35.py @@ -54,13 +54,14 @@ def build_model(): # Normalize characters using a dense layer model = Sequential() dense_layer = Dense(INPUT_VOCAB_SIZE, - input_shape=(INPUT_VOCAB_SIZE,)) + input_shape=(INPUT_VOCAB_SIZE,), + activation='softmax') model.add(dense_layer) - normalization_layer_set_weights(dense_layer) return model model = build_model() model.summary() +normalization_layer_set_weights(model.layers[0]) with open(sys.argv[1]) as f: for line in f: