schuetoliver
/
handson-ml
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

main
Aurélien Geron 2021-03-04 23:06:21 +13:00
parent 33f9ff10b4
commit e10a807907
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
07_ensemble_learning_and_random_forests.ipynb
View File

@ -327,11 +327,9 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"from math import ceil, sqrt\n",
"\n",
"bag_clf = BaggingClassifier(\n", "bag_clf = BaggingClassifier(\n",
" DecisionTreeClassifier(max_leaf_nodes=16),\n", " DecisionTreeClassifier(max_features=\"sqrt\", max_leaf_nodes=16),\n",
" n_estimators=500, max_features=ceil(sqrt(X_train.shape[1])), random_state=42)" " n_estimators=500, random_state=42)"
] ]
}, },
{ {