|
1467 | 1467 | "mglearn.discrete_scatter(X[:, 0], X[:, 1], y)\n",
|
1468 | 1468 | "line = np.linspace(-15, 15)\n",
|
1469 | 1469 | "for coef, intercept, color in zip(linear_svm.coef_, linear_svm.intercept_,\n",
|
1470 |
| - " ['b', 'r', 'g']):\n", |
| 1470 | + " mglearn.cm3.colors):\n", |
1471 | 1471 | " plt.plot(line, -(line * coef[0] + intercept) / coef[1], c=color)\n",
|
1472 | 1472 | "plt.ylim(-10, 15)\n",
|
1473 | 1473 | "plt.xlim(-10, 8)\n",
|
|
1511 | 1511 | "mglearn.discrete_scatter(X[:, 0], X[:, 1], y)\n",
|
1512 | 1512 | "line = np.linspace(-15, 15)\n",
|
1513 | 1513 | "for coef, intercept, color in zip(linear_svm.coef_, linear_svm.intercept_,\n",
|
1514 |
| - " ['b', 'r', 'g']):\n", |
| 1514 | + " mglearn.cm3.colors):\n", |
1515 | 1515 | " plt.plot(line, -(line * coef[0] + intercept) / coef[1], c=color)\n",
|
1516 | 1516 | "plt.legend(['Class 0', 'Class 1', 'Class 2', 'Line class 0', 'Line class 1',\n",
|
1517 | 1517 | " 'Line class 2'], loc=(1.01, 0.3))\n",
|
|
0 commit comments