Fix early stopping example, fixes #190

main
Aurélien Geron 2020-07-06 20:46:48 +12:00
parent df651cafb6
commit 05622d26b8
1 changed files with 2 additions and 3 deletions

View File

@ -845,7 +845,7 @@
}, },
"outputs": [], "outputs": [],
"source": [ "source": [
"from sklearn.base import clone\n", "from copy import deepcopy\n",
"\n", "\n",
"poly_scaler = Pipeline([\n", "poly_scaler = Pipeline([\n",
" (\"poly_features\", PolynomialFeatures(degree=90, include_bias=False)),\n", " (\"poly_features\", PolynomialFeatures(degree=90, include_bias=False)),\n",
@ -868,8 +868,7 @@
" if val_error < minimum_val_error:\n", " if val_error < minimum_val_error:\n",
" minimum_val_error = val_error\n", " minimum_val_error = val_error\n",
" best_epoch = epoch\n", " best_epoch = epoch\n",
" best_model = sgd_reg\n", " best_model = deepcopy(sgd_reg)"
" sgd_reg = clone(sgd_reg)"
] ]
}, },
{ {