Added colorblind capability

examples/mixture/plot_gmm.py

@@ -46,7 +46,8 @@
 dpgmm = mixture.DPGMM(n_components=5, covariance_type='full')
 dpgmm.fit(X)
 
-color_iter = itertools.cycle(['r', 'g', 'b', 'c', 'm'])
+color_iter = itertools.cycle(['navy', 'c', 'cornflowerblue', 'gold',
+                              'darkorange'])
 
 for i, (clf, title) in enumerate([(gmm, 'GMM'),
                                   (dpgmm, 'Dirichlet Process GMM')]):

examples/mixture/plot_gmm_classifier.py

@@ -38,8 +38,9 @@
 from sklearn.mixture import GMM
 
 
+colors = ['navy', 'turquoise', 'darkorange']
 def make_ellipses(gmm, ax):
-    for n, color in enumerate('rgb'):
+    for n, color in enumerate(colors):
         v, w = np.linalg.eigh(gmm._get_covars()[n][:2, :2])
         u = w[0] / np.linalg.norm(w[0])
         angle = np.arctan2(u[1], u[0])
@@ -91,14 +92,15 @@ def make_ellipses(gmm, ax):
     h = plt.subplot(2, n_classifiers / 2, index + 1)
     make_ellipses(classifier, h)
 
-    for n, color in enumerate('rgb'):
+    for n, color in enumerate(colors):
         data = iris.data[iris.target == n]
-        plt.scatter(data[:, 0], data[:, 1], 0.8, color=color,
+        plt.scatter(data[:, 0], data[:, 1], s=0.8, color=color,
                     label=iris.target_names[n])
     # Plot the test data with crosses
-    for n, color in enumerate('rgb'):
+    for n, color in enumerate(colors):
         data = X_test[y_test == n]
-        plt.plot(data[:, 0], data[:, 1], 'x', color=color)
+        print(color)
+        plt.scatter(data[:, 0], data[:, 1], marker='x', color=color)
 
     y_train_pred = classifier.predict(X_train)
     train_accuracy = np.mean(y_train_pred.ravel() == y_train.ravel()) * 100

examples/mixture/plot_gmm_selection.py

@@ -49,7 +49,8 @@
             best_gmm = gmm
 
 bic = np.array(bic)
-color_iter = itertools.cycle(['k', 'r', 'g', 'b', 'c', 'm', 'y'])
+color_iter = itertools.cycle(['navy', 'turquoise', 'cornflowerblue',
+                              'darkorange'])
 clf = best_gmm
 bars = []
 

examples/mixture/plot_gmm_sin.py

@@ -37,9 +37,8 @@
     X[i, 0] = x + np.random.normal(0, 0.1)
     X[i, 1] = 3 * (np.sin(x) + np.random.normal(0, .2))
 
-
-color_iter = itertools.cycle(['r', 'g', 'b', 'c', 'm'])
-
+color_iter = itertools.cycle(['navy', 'turquoise', 'cornflowerblue',
+                              'darkorange'])
 
 for i, (clf, title) in enumerate([
         (mixture.GMM(n_components=10, covariance_type='full', n_iter=100),
@@ -63,7 +62,7 @@
         # components.
         if not np.any(Y_ == i):
             continue
-        plt.scatter(X[Y_ == i, 0], X[Y_ == i, 1], .8, color=color)
+        plt.scatter(X[Y_ == i, 0], X[Y_ == i, 1], color=color, s=4)
 
         # Plot an ellipse to show the Gaussian component
         angle = np.arctan(u[1] / u[0])