"sgl-kernel/vscode:/vscode.git/clone" did not exist on "b7f3fec13c0280d341096a2db302f81ba4dbaf74"
Unverified Commit 887bbcb9 authored by Mark Daoust's avatar Mark Daoust Committed by GitHub
Browse files

Merge pull request #4943 from DecentGradient/patch-2

Explain why input_shape is needed here.
parents af41cc00 1be8e32a
......@@ -294,6 +294,7 @@
"cell_type": "code",
"source": [
"baseline_model = keras.Sequential([\n",
" # `input_shape` is only required here so that `.summary` works. \n"
" keras.layers.Dense(16, activation=tf.nn.relu, input_shape=(NUM_WORDS,)),\n",
" keras.layers.Dense(16, activation=tf.nn.relu),\n",
" keras.layers.Dense(1, activation=tf.nn.sigmoid)\n",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment