diff --git a/Notebooks/Chap05/5_1_Least_Squares_Loss.ipynb b/Notebooks/Chap05/5_1_Least_Squares_Loss.ipynb index d4281ee..890ce40 100644 --- a/Notebooks/Chap05/5_1_Least_Squares_Loss.ipynb +++ b/Notebooks/Chap05/5_1_Least_Squares_Loss.ipynb @@ -539,8 +539,8 @@ "# Hopefully, you can see that the maximum of the likelihood fn is at the same position as the minimum negative log likelihood\n", "# The least squares solution does not depend on sigma, so it's just flat -- no use here.\n", "# Let's check that:\n", - "print(\"Maximum likelihood = %3.3f, at beta_1=%3.3f\"%( (likelihoods[np.argmax(likelihoods)],sigma_vals[np.argmax(likelihoods)])))\n", - "print(\"Minimum negative log likelihood = %3.3f, at beta_1=%3.3f\"%( (nlls[np.argmin(nlls)],sigma_vals[np.argmin(nlls)])))\n", + "print(\"Maximum likelihood = %3.3f, at sigma=%3.3f\"%( (likelihoods[np.argmax(likelihoods)],sigma_vals[np.argmax(likelihoods)])))\n", + "print(\"Minimum negative log likelihood = %3.3f, at sigma=%3.3f\"%( (nlls[np.argmin(nlls)],sigma_vals[np.argmin(nlls)])))\n", "# Plot the best model\n", "sigma= sigma_vals[np.argmin(nlls)]\n", "y_model = shallow_nn(x_model, beta_0, omega_0, beta_1, omega_1)\n", @@ -564,4 +564,4 @@ } } ] -} \ No newline at end of file +}