Merge pull request #20 from GarrettJenkinson/patch-1

Fixing wide-and-deep to use normalized inputs
main
Aurélien Geron 2022-09-27 16:12:00 +13:00 committed by GitHub
commit 300e697004
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -1569,7 +1569,7 @@
"normalized = normalization_layer(input_)\n", "normalized = normalization_layer(input_)\n",
"hidden1 = hidden_layer1(normalized)\n", "hidden1 = hidden_layer1(normalized)\n",
"hidden2 = hidden_layer2(hidden1)\n", "hidden2 = hidden_layer2(hidden1)\n",
"concat = concat_layer([input_, hidden2])\n", "concat = concat_layer([normalized, hidden2])\n",
"output = output_layer(concat)\n", "output = output_layer(concat)\n",
"\n", "\n",
"model = tf.keras.Model(inputs=[input_], outputs=[output])" "model = tf.keras.Model(inputs=[input_], outputs=[output])"