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

Improve colors of diagrams min_samples_leaf_plot and sensitivity_to_rotation_plot

main
Aurélien Geron 2021-11-10 17:59:49 +13:00
parent 77da622124
commit 253e4e12a8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
05_decision_trees.ipynb
View File

@ -363,11 +363,11 @@
" \n",
" plt.contourf(x1, x2, y_pred, alpha=0.3, cmap=cmap)\n",
" plt.contour(x1, x2, y_pred, cmap=\"Greys\", alpha=0.8)\n",
" colors = {\"Wistia\": [\"blue\", \"black\"], \"Pastel1\": [\"red\", \"blue\"]}[cmap]\n",
" markers = (\"s\", \"^\")\n",
" colors = {\"Wistia\": [\"#78785c\", \"#c47b27\"], \"Pastel1\": [\"red\", \"blue\"]}\n",
" markers = (\"o\", \"^\")\n",
" for idx in (0, 1):\n",
" plt.plot(X[:, 0][y == idx], X[:, 1][y == idx],\n",
" color=colors[idx], marker=markers[idx], linestyle=\"none\")\n",
" color=colors[cmap][idx], marker=markers[idx], linestyle=\"none\")\n",
" plt.axis(axes)\n",
" plt.xlabel(r\"$x_1$\")\n",
" plt.ylabel(r\"$x_2$\", rotation=0)\n",