ENHanced the multilabel example aspect

vene · GaelVaroquaux · commit 2f79cb2c813f · 2012-01-10T23:34:13.000+01:00
diff --git a/examples/plot_multilabel.py b/examples/plot_multilabel.py
@@ -41,7 +41,7 @@ def plot_hyperplane(clf, min_x, max_x, linestyle, label):
     # get the separating hyperplane
     w = clf.coef_[0]
     a = -w[0] / w[1]
-    xx = np.linspace(min_x, max_x)
+    xx = np.linspace(min_x - 5, max_x + 5)  # make sure the line is long enough
     yy = a * xx - (clf.intercept_[0]) / w[1]
     pl.plot(xx, yy, linestyle, label=label)
 
@@ -81,27 +81,28 @@ def plot_subfigure(X, Y, subplot, title, transform):
     pl.xticks(())
     pl.yticks(())
 
-    if subplot == 1:
+    if subplot == 2:
+        pl.xlim(min_x - 5, max_x)
         pl.xlabel('First principal component')
         pl.ylabel('Second principal component')
-        pl.legend(loc="upper right")
+        pl.legend(loc="upper left")
 
 
-pl.figure(figsize=(13, 6))
+pl.figure(figsize=(8, 6))
 
 X, Y = make_multilabel_classification(n_classes=2, n_labels=1,
                                       allow_unlabeled=True,
-                                      random_state=0)
+                                      random_state=1)
 
 plot_subfigure(X, Y, 1, "With unlabeled samples + CCA", "cca")
 plot_subfigure(X, Y, 2, "With unlabeled samples + PCA", "pca")
 
 X, Y = make_multilabel_classification(n_classes=2, n_labels=1,
                                       allow_unlabeled=False,
-                                      random_state=0)
+                                      random_state=1)
 
 plot_subfigure(X, Y, 3, "Without unlabeled samples + CCA", "cca")
 plot_subfigure(X, Y, 4, "Without unlabeled samples + PCA", "pca")
 
-pl.subplots_adjust(.04, .07, .97, .90, .09, .2)
+pl.subplots_adjust(.04, .02, .97, .94, .09, .2)
 pl.show()
