Fix early stopping example, fixes ageron#190

FunnyRainn · Jul 6, 2020 · fabd609 · fabd609
1 parent 3426108
commit fabd609
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb
@@ -1363,7 +1363,8 @@
     "    if val_error < minimum_val_error:\n",
     "        minimum_val_error = val_error\n",
     "        best_epoch = epoch\n",
-    "        best_model = clone(sgd_reg)"
+    "        best_model = sgd_reg\n",
+    "        sgd_reg = clone(sgd_reg)"
    ]
   },
   {