Set max_features="sqrt" in Decision Tree rather than BaggingClassifier

2021-03-04 23:06:21 +13:00 · 2021-03-04 23:06:21 +13:00 · e10a807907
parent 33f9ff10b4
commit e10a807907
1 changed files with 2 additions and 4 deletions
--- a/07_ensemble_learning_and_random_forests.ipynb
+++ b/07_ensemble_learning_and_random_forests.ipynb
@ -327,11 +327,9 @@
   "metadata": {},
   "outputs": [],
   "source": [
-    "from math import ceil, sqrt\n",
-    "\n",
    "bag_clf = BaggingClassifier(\n",
-    "    DecisionTreeClassifier(max_leaf_nodes=16),\n",
-    "    n_estimators=500, max_features=ceil(sqrt(X_train.shape[1])), random_state=42)"
+    "    DecisionTreeClassifier(max_features=\"sqrt\", max_leaf_nodes=16),\n",
+    "    n_estimators=500, random_state=42)"
   ]
  },
  {