schuetoliver
/
handson-ml
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

SGD now defaults to lr=0.01 so use 1e-3 explicitly

main
Aurélien Geron 2019-06-10 10:53:32 +08:00
parent 3db31444cd
commit 400920f0aa
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
13_loading_and_preprocessing_data.ipynb
View File

@ -530,7 +530,7 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"model.compile(loss=\"mse\", optimizer=\"sgd\")" "model.compile(loss=\"mse\", optimizer=keras.optimizers.SGD(lr=1e-3))"
] ]
}, },
{ {
@ -1511,7 +1511,9 @@
" keras.layers.DenseFeatures(feature_columns=columns_without_target),\n", " keras.layers.DenseFeatures(feature_columns=columns_without_target),\n",
" keras.layers.Dense(1)\n", " keras.layers.Dense(1)\n",
"])\n", "])\n",
"model.compile(loss=\"mse\", optimizer=\"sgd\", metrics=[\"accuracy\"])\n", "model.compile(loss=\"mse\",\n",
" optimizer=keras.optimizers.SGD(lr=1e-3),\n",
" metrics=[\"accuracy\"])\n",
"model.fit(dataset, steps_per_epoch=len(X_train) // batch_size, epochs=5)" "model.fit(dataset, steps_per_epoch=len(X_train) // batch_size, epochs=5)"
] ]
}, },
@ -1635,7 +1637,9 @@
" keras.layers.Flatten(input_shape=[28, 28, 1]),\n", " keras.layers.Flatten(input_shape=[28, 28, 1]),\n",
" keras.layers.Lambda(lambda images: tf.cast(images, tf.float32)),\n", " keras.layers.Lambda(lambda images: tf.cast(images, tf.float32)),\n",
" keras.layers.Dense(10, activation=\"softmax\")])\n", " keras.layers.Dense(10, activation=\"softmax\")])\n",
"model.compile(loss=\"sparse_categorical_crossentropy\", optimizer=\"sgd\", metrics=[\"accuracy\"])\n", "model.compile(loss=\"sparse_categorical_crossentropy\",\n",
" optimizer=keras.optimizers.SGD(lr=1e-3),\n",
" metrics=[\"accuracy\"])\n",
"model.fit(mnist_train, steps_per_epoch=60000 // 32, epochs=5)" "model.fit(mnist_train, steps_per_epoch=60000 // 32, epochs=5)"
] ]
}, },