diff --git a/Notebooks/Chap06/6_4_Momentum.ipynb b/Notebooks/Chap06/6_4_Momentum.ipynb index de38182..e7f7c9d 100644 --- a/Notebooks/Chap06/6_4_Momentum.ipynb +++ b/Notebooks/Chap06/6_4_Momentum.ipynb @@ -61,7 +61,7 @@ { "cell_type": "code", "source": [ - "# Let's create our training data 30 pairs {x_i, y_i}\n", + "# Let's create our training data of 30 pairs {x_i, y_i}\n", "# We'll try to fit the Gabor model to these data\n", "data = np.array([[-1.920e+00,-1.422e+01,1.490e+00,-1.940e+00,-2.389e+00,-5.090e+00,\n", " -8.861e+00,3.578e+00,-6.010e+00,-6.995e+00,3.634e+00,8.743e-01,\n", @@ -137,7 +137,7 @@ { "cell_type": "markdown", "source": [ - "Now lets compute the sum of squares loss for the training data and plot the loss function" + "Now let's compute the sum of squares loss for the training data and plot the loss function" ], "metadata": { "id": "QU5mdGvpTtEG" @@ -160,7 +160,7 @@ " b = np.floor(my_colormap_vals_dec - r * 256 *256 - g * 256)\n", " my_colormap = ListedColormap(np.vstack((r,g,b)).transpose()/255.0)\n", "\n", - " # Make grid of intercept/slope values to plot\n", + " # Make grid of offset/frequency values to plot\n", " offsets_mesh, freqs_mesh = np.meshgrid(np.arange(-10,10.0,0.1), np.arange(2.5,22.5,0.1))\n", " loss_mesh = np.zeros_like(freqs_mesh)\n", " # Compute loss for every set of parameters\n", @@ -365,7 +365,6 @@ "\n", " # Update the parameters\n", " phi_all[:,c_step+1:c_step+2] = phi_all[:,c_step:c_step+1] - alpha * momentum\n", - " # Measure loss and draw model every 8th step\n", "\n", "loss = compute_loss(data[0,:], data[1,:], model, phi_all[:,c_step+1:c_step+2])\n", "draw_model(data,model,phi_all[:,c_step+1], \"Iteration %d, loss = %f\"%(c_step+1,loss))\n", @@ -387,4 +386,4 @@ } } ] -} \ No newline at end of file +}