[MRG] Modify Classification module example matplotlibv2
@@ -95,9 +95,11 @@ for ds_cnt, ds in enumerate(datasets):
if ds_cnt == 0:
ax.set_title("Input data")
# Plot the training points
- ax.scatter(X_train[:, 0], X_train[:, 1], c=y_train, cmap=cm_bright)
+ ax.scatter(X_train[:, 0], X_train[:, 1], c=y_train, cmap=cm_bright,
+ edgecolors='k')
# and testing points
- ax.scatter(X_test[:, 0], X_test[:, 1], c=y_test, cmap=cm_bright, alpha=0.6)
+ ax.scatter(X_test[:, 0], X_test[:, 1], c=y_test, cmap=cm_bright, alpha=0.6,
+ edgecolors='k')
ax.set_xlim(xx.min(), xx.max())
ax.set_ylim(yy.min(), yy.max())
ax.set_xticks(())
@@ -122,10 +124,11 @@ for ds_cnt, ds in enumerate(datasets):
ax.contourf(xx, yy, Z, cmap=cm, alpha=.8)
# Plot also the training points
- ax.scatter(X_train[:, 0], X_train[:, 1], c=y_train, cmap=cm_bright)
+ ax.scatter(X_train[:, 0], X_train[:, 1], c=y_train, cmap=cm_bright,
+ edgecolors='k')
# and testing points
ax.scatter(X_test[:, 0], X_test[:, 1], c=y_test, cmap=cm_bright,
- alpha=0.6)
+ edgecolors='k', alpha=0.6)
ax.set_xlim(xx.min(), xx.max())
ax.set_ylim(yy.min(), yy.max())
