Clarify error message for min_samples_split. (scikit-learn#8167)

mikebenfield · maskani-moh · commit c4fc08c9374f · 2017-11-15T12:11:36.000-05:00
diff --git a/sklearn/tree/tree.py b/sklearn/tree/tree.py
@@ -185,14 +185,16 @@ def fit(self, X, y, sample_weight=None, check_input=True,
 
         if isinstance(self.min_samples_split, (numbers.Integral, np.integer)):
             if not 2 <= self.min_samples_split:
-                raise ValueError("min_samples_split must be at least 2 "
-                                 "or in (0, 1.0], got %s"
+                raise ValueError("min_samples_split must be an integer "
+                                 "greater than 1 or a float in (0.0, 1.0]; "
+                                 "got the integer %s"
                                  % self.min_samples_split)
             min_samples_split = self.min_samples_split
         else:  # float
             if not 0. < self.min_samples_split <= 1.:
-                raise ValueError("min_samples_split must be at least 2 "
-                                 "or in (0, 1.0], got %s"
+                raise ValueError("min_samples_split must be an integer "
+                                 "greater than 1 or a float in (0.0, 1.0]; "
+                                 "got the float %s"
                                  % self.min_samples_split)
             min_samples_split = int(ceil(self.min_samples_split * n_samples))
             min_samples_split = max(2, min_samples_split)
