add params for avoiding warn and improving perf.

2026-01-14 12:14:36 +01:00 · 2018-01-30 17:19:37 +09:00
parent 0f46b85009
commit 385d635e92
3 changed files with 8 additions and 8 deletions
--- a/02_end_to_end_machine_learning_project.ipynb
+++ b/02_end_to_end_machine_learning_project.ipynb
@@ -1503,7 +1503,7 @@
    "forest_reg = RandomForestRegressor(random_state=42)\n",
    "# train across 5 folds, that's a total of (12+6)*5=90 rounds of training \n",
    "grid_search = GridSearchCV(forest_reg, param_grid, cv=5,\n",
-    "                           scoring='neg_mean_squared_error')\n",
+    "                           scoring='neg_mean_squared_error', return_train_score=True)\n",
    "grid_search.fit(housing_prepared, housing_labels)"
   ]
  },