Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
samples/core/tutorials/keras/overfit_and_underfit.ipynb
View file @ 887bbcb9
...@@ -294,6 +294,7 @@ ...@@ -294,6 +294,7 @@
"cell_type": "code", "cell_type": "code",
"source": [ "source": [
"baseline_model = keras.Sequential([\n", "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, input_shape=(NUM_WORDS,)),\n",
" keras.layers.Dense(16, activation=tf.nn.relu),\n", " keras.layers.Dense(16, activation=tf.nn.relu),\n",
" keras.layers.Dense(1, activation=tf.nn.sigmoid)\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