use learning rate as func parameter

rasbt · Jun 30, 2018 · 9c9e714 · 9c9e714
1 parent 7808f9f
commit 9c9e714
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/code/ch15/ch15.ipynb b/code/ch15/ch15.ipynb
@@ -864,7 +864,7 @@
    },
    "outputs": [],
    "source": [
-    "def build_cnn():\n",
+    "def build_cnn(learning_rate=1e-4):\n",
     "    ## Placeholders for X and y:\n",
     "    tf_x = tf.placeholder(tf.float32, shape=[None, 784],\n",
     "                          name='tf_x')\n",
@@ -1056,8 +1056,7 @@
     "import tensorflow as tf\n",
     "import numpy as np\n",
     "\n",
-    "## Define hyperparameters\n",
-    "learning_rate = 1e-4\n",
+    "## Define random seed\n",
     "random_seed = 123\n",
     "\n",
     "np.random.seed(random_seed)\n",