Cleaning

scikit-learn · Oct 20, 2015 · 610c6a0 · 610c6a0
1 parent f16c9ae
commit 610c6a0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sklearn/utils/estimator_checks.py b/sklearn/utils/estimator_checks.py
@@ -1029,11 +1029,13 @@ def check_classifiers_train(name, Classifier, readonly=False):
         positive = True
     else:
         positive = False
-    X_m, y_m = _make_blobs_with_mode(random_state=0, shuffle=True,
+    X_m, y_m = _make_blobs_with_mode(n_samples=300,
+                                     random_state=0, shuffle=True,
                                      readonly=readonly, scale=True,
                                      positive=positive)
     # generate binary problem
-    X_b, y_b = _make_blobs_with_mode(random_state=0, shuffle=True,
+    X_b, y_b = _make_blobs_with_mode(n_samples=300,
+                                     random_state=0, shuffle=True,
                                      readonly=readonly, scale=True,
                                      positive=positive, centers=2)
     for (X, y) in [(X_m, y_m), (X_b, y_b)]: