Merge pull request #20 from GarrettJenkinson/patch-1
Fixing wide-and-deep to use normalized inputsmain
commit
300e697004
|
@ -1569,7 +1569,7 @@
|
|||
"normalized = normalization_layer(input_)\n",
|
||||
"hidden1 = hidden_layer1(normalized)\n",
|
||||
"hidden2 = hidden_layer2(hidden1)\n",
|
||||
"concat = concat_layer([input_, hidden2])\n",
|
||||
"concat = concat_layer([normalized, hidden2])\n",
|
||||
"output = output_layer(concat)\n",
|
||||
"\n",
|
||||
"model = tf.keras.Model(inputs=[input_], outputs=[output])"
|
||||
|
|
Loading…
Reference in New Issue